yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers
MIT License
29.54k stars 8k forks source link

Hi, it's a problem in GAN #211

Closed xcc13 closed 4 years ago

xcc13 commented 4 years ago

create web directory ./checkpoints\experiment_name\web... Traceback (most recent call last): File "H:/Code/aaaaaaaaaa/pix2pix/train.py", line 76, in model.optimize_parameters() # calculate loss functions, get gradients, update network weights File "H:\Code\aaaaaaaaaa\pix2pix\models\Ea_GAN_model.py", line 176, in optimize_parameters self.backward_G() # calculate graidents for G File "H:\Code\aaaaaaaaaa\pix2pix\models\Ea_GAN_model.py", line 158, in backward_G self.loss_G.backward() File "D:\Python\Anaconda\envs\pytorch\lib\site-packages\torch\tensor.py", line 107, in backward torch.autograd.backward(self, gradient, retain_graph, create_graph) File "D:\Python\Anaconda\envs\pytorch\lib\site-packages\torch\autograd__init__.py", line 93, in backward allow_unreachable=True) # allow_unreachable flag RuntimeError: ntensor >= 3 ASSERT FAILED at aten\src\ATen\native\cpu\IndexKernel.cpp.AVX2.cpp:51, please report a bug to PyTorch.

Process finished with exit code 1

And with cuda:

create web directory ./checkpoints/0702_16/web... Traceback (most recent call last): File "/home/qian/XCC/Ea_GAN/train.py", line 76, in <module> model.optimize_parameters() # calculate loss functions, get gradients, update network weights File "/home/qian/XCC/Ea_GAN/models/Ea_GAN_model.py", line 179, in optimize_parameters self.backward_G() # calculate graidents for G File "/home/qian/XCC/Ea_GAN/models/Ea_GAN_model.py", line 161, in backward_G self.loss_G.backward() File "/home/qian/anaconda3/envs/XCC/lib/python3.7/site-packages/torch/tensor.py", line 107, in backward torch.autograd.backward(self, gradient, retain_graph, create_graph) File "/home/qian/anaconda3/envs/XCC/lib/python3.7/site-packages/torch/autograd/__init__.py", line 93, in backward allow_unreachable=True) # allow_unreachable flag RuntimeError: N <= iter.ntensors() ASSERT FAILED at /opt/conda/conda-bld/pytorch_1556653215914/work/aten/src/ATen/native/cuda/IndexKernel.cu:13, please report a bug to PyTorch.