yihong-chen / DREAM

rnn based model for recommendations
MIT License
89 stars 43 forks source link

TypeError: iteration over a 0-d tensor #10

Open tristaneljed opened 5 years ago

tristaneljed commented 5 years ago

Running the train file results in this error:

basket_pool_type max rnn_layers 2 rnn_type RNN_RELU dropout 0.5 num_product 49690 none_idx 49689 embedding_dim 128 cuda False clip 200 epochs 100 batch_size 32 learning_rate 0.001 log_interval 1 checkpoint_dir ../dream/dream-{epoch:02d}-{loss:.4f}.model


---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-1-423abe3e7cdf> in <module>()
270             train_reorder_dream()
271         else:
--> 272             train_dream()
273         print('-' * 89)
274         if constants.REORDER:
in train_dream() 82 for i, x in enumerate(batchify(train_ub, dr_config.batch_size)): 83 baskets, lens, _ = x ---> 84 dr_hidden = repackage_hidden(dr_hidden) # repackage hidden state for RNN 85 dr_model.zero_grad() # optim.zero_grad() 86 dynamic_user, _ = dr_model(baskets, lens, dr_hidden) ~/SageMaker/MBA/Instacart/src/utils.py in repackage_hidden(h) 92 return Variable(h.data) 93 else: ---> 94 return tuple(repackage_hidden(v) for v in h) 95 96 ###################### Summary ~/SageMaker/MBA/Instacart/src/utils.py in (.0) 92 return Variable(h.data) 93 else: ---> 94 return tuple(repackage_hidden(v) for v in h) 95 96 ###################### Summary ~/SageMaker/MBA/Instacart/src/utils.py in repackage_hidden(h) 92 return Variable(h.data) 93 else: ---> 94 return tuple(repackage_hidden(v) for v in h) 95 96 ###################### Summary ~/SageMaker/MBA/Instacart/src/utils.py in (.0) 92 return Variable(h.data) 93 else: ---> 94 return tuple(repackage_hidden(v) for v in h) 95 96 ###################### Summary ~/SageMaker/MBA/Instacart/src/utils.py in repackage_hidden(h) 92 return Variable(h.data) 93 else: ---> 94 return tuple(repackage_hidden(v) for v in h) 95 96 ###################### Summary ~/SageMaker/MBA/Instacart/src/utils.py in (.0) 92 return Variable(h.data) 93 else: ---> 94 return tuple(repackage_hidden(v) for v in h) 95 96 ###################### Summary ~/SageMaker/MBA/Instacart/src/utils.py in repackage_hidden(h) 92 return Variable(h.data) 93 else: ---> 94 return tuple(repackage_hidden(v) for v in h) 95 96 ###################### Summary ~/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/tensor.py in __iter__(self) 382 # map will interleave them.) 383 if self.dim() == 0: --> 384 raise TypeError('iteration over a 0-d tensor') 385 if torch._C._get_tracing_state(): 386 warnings.warn('Iterating over a tensor might cause the trace to be incorrect. ' TypeError: iteration over a 0-d tensor ```
xumanru commented 2 years ago

did you solve it?

patcharees commented 2 years ago

I got the same problem. Any updates?