tb.py

Traceback (most recent call last):
  File "borrar.py", line 47, in <module>
    decoder_dist(input_[:,c].contiguous(), hidden) #RuntimeError: Expected hidden size (1, 4, 64), got (1, 32, 64)
  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 56, in forward
    inputs, kwargs = self.scatter(inputs, kwargs, self.device_ids)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 67, in scatter
    return scatter_kwargs(inputs, kwargs, device_ids, dim=self.dim)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 30, in scatter_kwargs
    inputs = scatter(inputs, target_gpus, dim)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 25, in scatter
    return scatter_map(inputs)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 18, in scatter_map
    return tuple(zip(*map(scatter_map, obj)))
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 15, in scatter_map
    return Scatter(target_gpus, dim=dim)(obj)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/nn/parallel/_functions.py", line 60, in forward
    outputs = comm.scatter(input, self.target_gpus, self.chunk_sizes, self.dim, streams)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/cuda/comm.py", line 145, in scatter
    chunks = tensor.chunk(len(devices), dim)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/tensor.py", line 174, in chunk
    return torch.chunk(self, n_chunks, dim)
  File "/home/ubuntu/miniconda3/envs/torchenv/lib/python3.6/site-packages/torch/functional.py", line 44, in chunk
    split_size = (tensor.size(dim) + chunks - 1) // chunks
RuntimeError: invalid argument 2: out of range at /opt/conda/conda-bld/pytorch_1502009910772/work/torch/lib/THC/generic/THCTensor.c:23