Hello everyone, I have my own loss function, and it works well before I upgrade pytorch to the latest version by following the post Updating PyTorch, then, my code does not work anymore, there is no warning or error but the program is stuck at the 146-th line of “variable.py”

self._execution_engine.run_backward((self,), (gradient,), retain_variables)

I even cannot step into it, and if I stop the program manually, it always shows

Traceback (most recent call last):
File “/home/lii/anaconda2/lib/python2.7/multiprocessing/process.py”, line 258, in _bootstrap
self.run()
File “/home/lii/anaconda2/lib/python2.7/multiprocessing/process.py”, line 114, in run
self._target(*self._args, **self._kwargs)
File “/home/lii/anaconda2/lib/python2.7/site-packages/torch/utils/data/dataloader.py”, line 28, in _worker_loop
r = index_queue.get()
File “/home/lii/anaconda2/lib/python2.7/multiprocessing/queues.py”, line 378, in get
return recv()
File “/home/lii/anaconda2/lib/python2.7/site-packages/torch/multiprocessing/queue.py”, line 21, in recv
buf = self.recv_bytes()
KeyboardInterrupt

I am using anaconda2 in my workstation with CPU only, but the same code works well in GPU on a server with old pytorch version. So, could I downgrade pytorch or should I do something more?
Thank you!