diff --git a/checkpoints/ROMA_UNSB_001/loss_log.txt b/checkpoints/ROMA_UNSB_001/loss_log.txt
index 09b323f..8d9f361 100644
--- a/checkpoints/ROMA_UNSB_001/loss_log.txt
+++ b/checkpoints/ROMA_UNSB_001/loss_log.txt
@@ -5,3 +5,4 @@
 ================ Training Loss (Sun Feb 23 16:05:19 2025) ================
 ================ Training Loss (Sun Feb 23 16:06:44 2025) ================
 ================ Training Loss (Sun Feb 23 16:09:38 2025) ================
+================ Training Loss (Sun Feb 23 16:44:56 2025) ================
diff --git a/models/__pycache__/networks.cpython-39.pyc b/models/__pycache__/networks.cpython-39.pyc
index cb079a1..800e7d3 100644
Binary files a/models/__pycache__/networks.cpython-39.pyc and b/models/__pycache__/networks.cpython-39.pyc differ
diff --git a/models/__pycache__/roma_unsb_model.cpython-39.pyc b/models/__pycache__/roma_unsb_model.cpython-39.pyc
index f14a8b5..4914986 100644
Binary files a/models/__pycache__/roma_unsb_model.cpython-39.pyc and b/models/__pycache__/roma_unsb_model.cpython-39.pyc differ
diff --git a/scripts/train.sh b/scripts/train.sh
index d005fe5..1c60f02 100755
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -29,4 +29,5 @@ python train.py \
     --flip_equivariance False \
     --eta_ratio 0.1 \
     --tau 0.1 \
-    --num_timesteps 10
+    --num_timesteps 10 \
+    --input_nc 1