Seanny123 / da-rnn

Dual-Stage Attention-Based Recurrent Neural Net for Time Series Prediction
333 stars 118 forks source link

RuntimeError #1

Open dosxp opened 6 years ago

dosxp commented 6 years ago

win7 64bit, python 3.7.0, cuda 9

Warning (from warnings module): File "E:\python370\lib\site-packages\sklearn\externals\joblib\externals\cloudpickle\cloudpickle.py", line 47 import imp DeprecationWarning: the imp module is deprecated in favour of importlib; see the module's documentation for alternative uses 2018-11-06 22:37:50,303 - VOC_TOPICS - INFO - Using computation device: cuda:0 2018-11-06 22:37:52,505 - VOC_TOPICS - INFO - Shape of data: (40560, 82). Missing in data: 0. 2018-11-06 22:37:52,637 - VOC_TOPICS - INFO - Training size: 28392. 2018-11-06 22:38:02,391 - VOC_TOPICS - INFO - Iterations per epoch: 221.812 ~ 222. Traceback (most recent call last): File "E:\python370\Scripts\da-rnn-master\main.py", line 196, in iter_loss, epoch_loss = train(model, data, config, n_epochs=10, save_plots=save_plots) File "E:\python370\Scripts\da-rnn-master\main.py", line 84, in train loss = train_iteration(net, t_cfg.loss_func, feats, y_history, y_target) File "E:\python370\Scripts\da-rnn-master\main.py", line 143, in train_iteration input_weighted, input_encoded = t_net.encoder(numpy_to_tvar(X)) File "E:\python370\lib\site-packages\torch\nn\modules\module.py", line 477, in call result = self.forward(*input, *kwargs) File "E:\python370\Scripts\da-rnn-master\modules.py", line 43, in forward input_data.permute(0, 2, 1)), dim=2) # batch_size input_size (2hidden_size + T - 1) RuntimeError: Expected a Tensor of type torch.FloatTensor but found a type torch.cuda.FloatTensor for sequence element 2 in sequence argument at position #1 'tensors'