fatchord / fftnet Goto Github PK

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-81-d644dae333d9> in <module>()
      3 #data = torch.zeros((50000,))
      4 print(data.shape)
----> 5 train(data, model, optimizer, batch_size=2, seq_len=5000, lr=1e-5, steps=10_000)

~/dev/FFTNet/fftnet.py in train(data, model, optimizer, batch_size, seq_len, lr, steps)
    138         print(y_hat[:, :, 1:])
    139         print(y.unsqueeze(-1))
--> 140         loss = criterion(y_hat[:, :, 1:], y.unsqueeze(-1))
    141 
    142         running_loss += loss.item()

/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py in __call__(self, *input, **kwargs)
    548             result = self._slow_forward(*input, **kwargs)
    549         else:
--> 550             result = self.forward(*input, **kwargs)
    551         for hook in self._forward_hooks.values():
    552             hook_result = hook(self, input, result)

/usr/local/lib/python3.7/site-packages/torch/nn/modules/loss.py in forward(self, input, target)
    203 
    204     def forward(self, input, target):
--> 205         return F.nll_loss(input, target, weight=self.weight, ignore_index=self.ignore_index, reduction=self.reduction)
    206 
    207 

/usr/local/lib/python3.7/site-packages/torch/nn/functional.py in nll_loss(input, target, weight, size_average, ignore_index, reduce, reduction)
   2115         ret = torch._C._nn.nll_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
   2116     elif dim == 4:
-> 2117         ret = torch._C._nn.nll_loss2d(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
   2118     else:
   2119         # dim == 3 or dim > 4

IndexError: Target -9 is out of bounds.