123456/roma_unsb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Traceback (most recent call last):

Browse Source 
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
This commit is contained in:
bishe 2025-02-23 16:49:26 +08:00
parent 9966211e8b
commit a798da6b32
4 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
checkpoints/ROMA_UNSB_001/loss_log.txt
View File

@ -5,3 +5,4 @@
================ Training Loss (Sun Feb 23 16:05:19 2025) ================ ================ Training Loss (Sun Feb 23 16:05:19 2025) ================
================ Training Loss (Sun Feb 23 16:06:44 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:09:38 2025) ================
================ Training Loss (Sun Feb 23 16:44:56 2025) ================

BIN
models/__pycache__/networks.cpython-39.pyc
View File

Binary file not shown.

BIN
models/__pycache__/roma_unsb_model.cpython-39.pyc
View File

Binary file not shown.

3
scripts/train.sh
View File

@ -29,4 +29,5 @@ python train.py \
--flip_equivariance False \ --flip_equivariance False \
--eta_ratio 0.1 \ --eta_ratio 0.1 \
--tau 0.1 \ --tau 0.1 \
--num_timesteps 10 --num_timesteps 10 \
--input_nc 1