Traceback (most recent call last):
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/intuitive_control_note.py", line 149, in
output_dict = net_G(data['source_image'].cuda(), data['target_semantics'].cuda())
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/generators/face_model.py", line 35, in forward
output = self.warpping_net(input_image, descriptor)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/generators/face_model.py", line 94, in forward
output = self.hourglass(input_image, descriptor)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/generators/base_function.py", line 44, in forward
return self.decoder(self.encoder(x, z), z)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/generators/base_function.py", line 200, in forward
out = torch.cat([out, x.pop()], 1) if self.skip_connect else out
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 88 but got size 89 for tensor number 1 in the list.
Traceback (most recent call last):
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/intuitive_control_note.py", line 149, in
output_dict = net_G(data['source_image'].cuda(), data['target_semantics'].cuda())
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/generators/face_model.py", line 35, in forward
output = self.warpping_net(input_image, descriptor)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/generators/face_model.py", line 94, in forward
output = self.hourglass(input_image, descriptor)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/generators/base_function.py", line 44, in forward
return self.decoder(self.encoder(x, z), z)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/Conda_dir/my_environment/said_312/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/media/ubuntu/1T1/ubuntu_data/nlp/demo_40_geneface/PIRender/generators/base_function.py", line 200, in forward
out = torch.cat([out, x.pop()], 1) if self.skip_connect else out
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 88 but got size 89 for tensor number 1 in the list.