carolineec/EverybodyDanceNow

requirements.txt ?

sairajd044 opened this issue · 0 comments

I'm getting this error while training

---------- Networks initialized ------------- C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\_compile.py:24: UserWarning: optimizer contains a parameter group with duplicate parameters; in future, this will cause an error; see github.com/pytorch/pytorch/issues/40967 for more information return torch._dynamo.disable(fn, recursive)(*args, **kwargs) model [Pix2PixHDModel] was created create web directory WHERE_TO_SAVE_CHECKPOINTS\MY_MODEL_NAME_global\web... Traceback (most recent call last): File "D:\MTP\EverybodyDanceNow\train_fullts.py", line 63, in <module> losses, generated = model(Variable(data['label']), Variable(data['next_label']), Variable(data['image']), \ File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl return forward_call(*args, **kwargs) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\parallel\data_parallel.py", line 183, in forward return self.module(*inputs[0], **module_kwargs[0]) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl return forward_call(*args, **kwargs) File "D:\MTP\EverybodyDanceNow\models\pix2pixHD_model_fullts.py", line 201, in forward I_0 = self.netG.forward(input_concat) File "D:\MTP\EverybodyDanceNow\models\networks.py", line 228, in forward return self.model(input) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl return forward_call(*args, **kwargs) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\container.py", line 217, in forward input = module(input) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl return forward_call(*args, **kwargs) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\instancenorm.py", line 87, in forward return self._apply_instance_norm(input) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\instancenorm.py", line 36, in _apply_instance_norm return F.instance_norm( File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\functional.py", line 2525, in instance_norm _verify_spatial_size(input.size()) File "C:\Users\saira\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\functional.py", line 2493, in _verify_spatial_size raise ValueError(f"Expected more than 1 spatial element when training, got input size {size}") ValueError: Expected more than 1 spatial element when training, got input size torch.Size([1, 512, 1, 1])

Perhaps it is because of different pytorch version. May I get a requirements.txt file to run correctly ?