File "train.py", line 76, in <module>
    main(config, args.resume)
  File "train.py", line 45, in main
    trainer.train()
  File "C:\Users\Casper\Desktop\TU\2019-2020\AnalysisMachineLearning\project\log-strategy\JackTheLoggerNet\base\base_trainer.py", line 88, in train
    result = self._train_epoch(epoch)
  File "C:\Users\Casper\Desktop\TU\2019-2020\AnalysisMachineLearning\project\log-strategy\JackTheLoggerNet\trainer\trainer.py", line 52, in _train_epoch
    loss = self.loss(output, target)
  File "C:\Users\Casper\Desktop\TU\2019-2020\AnalysisMachineLearning\project\log-strategy\JackTheLoggerNet\model\loss.py", line 6, in cross_entropy_loss
    return F.cross_entropy(output, target, weight=torch.tensor([1.,25.]))
  File "C:\ProgramData\Miniconda3\lib\site-packages\torch\nn\functional.py", line 2009, in cross_entropy
    return nll_loss(log_softmax(input, 1), target, weight, None, ignore_index, None, reduction)
  File "C:\ProgramData\Miniconda3\lib\site-packages\torch\nn\functional.py", line 1838, in nll_loss
    ret = torch._C._nn.nll_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
RuntimeError: Expected object of device type cuda but got device type cpu for argument #3 'weight' in call to _thnn_nll_loss_forward