there is no gradient
Hi author, good job.when i ran the train.py,i meet this error.
File "D:\mycodeapp\anaconda\envs\py7\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\mycode\deep-stabilization-master\dvs\model.py", line 27, in forward
self.hx, self.cx = self.LSTM(x, (self.hx, self.cx))
File "D:\mycodeapp\anaconda\envs\py7\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\mycodeapp\anaconda\envs\py7\lib\site-packages\torch\nn\modules\rnn.py", line 974, in forward
self.bias_ih, self.bias_hh,
(function print_stack)
Traceback (most recent call last):
File "D:/mycode/deep-stabilization-master/dvs/train.py", line 276, in
Hi! did you find a solution to your problem? I also had the same problem.
It seems the problem of the Pytorch version. Please try Pytorch 1.0.0 or replace LayerLSTM class with nn.LSTM.
It seems the problem of the Pytorch version. Please try Pytorch 1.0.0 or replace
LayerLSTMclass withnn.LSTM.
Problem solved! Thanks for your reply!
hello, could you tell me how to replace LayerLSTM with nn.LSTM