t.py

Traceback (most recent call last):                                                                                                                 
  File "borrar.py", line 45, in <module>                                                                                                           
    decoder_dist(input_[:,c].contiguous(), hidden)                                                                                                 
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/modules/module.py", line 224, in __call__                       
    result = self.forward(*input, **kwargs)                                                                                                        
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 60, in forward                 
    outputs = self.parallel_apply(replicas, inputs, kwargs)                                                                                        
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 70, in parallel_apply          
    return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])                                                               
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 67, in parallel_apply
    raise output
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 42, in _worker
    output = module(*input, **kwargs)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/modules/module.py", line 224, in __call__
    result = self.forward(*input, **kwargs)
  File "borrar.py", line 21, in forward
    output, hidden = self.rnn(encoded.view(1, batch_size, -1), hidden)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/modules/module.py", line 224, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/modules/rnn.py", line 162, in forward
    output, hidden = func(input, self.all_weights, hx)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/_functions/rnn.py", line 351, in forward
    return func(input, *fargs, **fkwargs)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/autograd/function.py", line 284, in _do_forward
    flat_output = super(NestedIOFunction, self)._do_forward(*flat_input)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/autograd/function.py", line 306, in forward
    result = self.forward_extended(*nested_tensors)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/_functions/rnn.py", line 293, in forward_extended
    cudnn.rnn.forward(self, input, hx, weight, output, hy)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/backends/cudnn/rnn.py", line 266, in forward
    hidden_size, tuple(hx.size())))
RuntimeError: Expected hidden size (1, 2, 8), got (1, 16, 8)