Set_storage is not allowed

Hi,

I’m receiving follow error when I try to train model on multi-GPU. Can someone let me know what’s wrong?

Thanks.


Traceback (most recent call last):
File “train.py”, line 247, in
out, output_sizes = model(inputs, input_sizes)
File “/home/edward/anaconda3/envs/deepspeech2_p36/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 491, in call
result = self.forward(*input, **kwargs)
File “/home/edward/anaconda3/envs/deepspeech2_p36/lib/python3.6/site-packages/torch/nn/parallel/distributed.py”, line 360, in forward
self._sync_params()
File “/home/edward/anaconda3/envs/deepspeech2_p36/lib/python3.6/site-packages/torch/nn/parallel/distributed.py”, line 392, in sync_params
param_data.set
(tensor)
RuntimeError: set_storage is not allowed on Tensor created from .data or .detach()