Traceback (most recent call last):

File "/home/openxs/jj/roma_unsb/train.py", line 47, in <module>
    model.optimize_parameters()   # calculate loss functions, get gradients, update network weights
  File "/home/openxs/jj/roma_unsb/models/roma_unsb_model.py", line 315, in optimize_parameters
    self.forward()
  File "/home/openxs/jj/roma_unsb/models/roma_unsb_model.py", line 445, in forward
    Xt_1 = self.netG(Xt, self.time, z)
  File "/home/openxs/anaconda3/envs/I2V/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/openxs/anaconda3/envs/I2V/lib/python3.9/site-packages/torch/nn/parallel/data_parallel.py", line 169, in forward
    return self.module(*inputs[0], **kwargs[0])
  File "/home/openxs/anaconda3/envs/I2V/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/openxs/jj/roma_unsb/models/networks.py", line 980, in forward
    feat = layer(feat)
  File "/home/openxs/anaconda3/envs/I2V/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/openxs/anaconda3/envs/I2V/lib/python3.9/site-packages/torch/nn/modules/conv.py", line 463, in forward
    return self._conv_forward(input, self.weight, self.bias)
  File "/home/openxs/anaconda3/envs/I2V/lib/python3.9/site-packages/torch/nn/modules/conv.py", line 459, in _conv_forward
    return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Given groups=1, weight of size [64, 3, 7, 7], expected input[1, 1, 1734, 774] to have 3 channels, but got 1 channels instead

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) ================

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