-
Notifications
You must be signed in to change notification settings - Fork 87
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Runtime error during 'python train.py' #8
Comments
delete .squeeze(2) |
Does not seem to help preprocessed data was found and loaded |
That's a different error. What I did for this was just always return True in parameters_allocation_check function. |
@ruotianluo which version of pytorch do you working on? |
I successfully ran the word embedding, but while training I get this runtime error. Any suggestions?
File "train.py", line 59, in
cross_entropy, kld, coef = train_step(iteration, args.batch_size, args.use_cuda, args.dropout)
File "
/pytorch_RVAE/model/rvae.py", line 104, in train/pytorch/torch/lib/python2.7/site-packages/torch/nn/modules/module.py", line 325, in callz=None)
File "
result = self.forward(*input, **kwargs)
File "
/pytorch_RVAE/model/rvae.py", line 64, in forward/pytorch/torch/lib/python2.7/site-packages/torch/nn/modules/module.py", line 325, in callencoder_input = self.embedding(encoder_word_input, encoder_character_input)
File "/
result = self.forward(*input, **kwargs)
File "
/pytorch_RVAE/selfModules/embedding.py", line 47, in forward/pytorch/torch/lib/python2.7/site-packages/torch/nn/modules/module.py", line 325, in callcharacter_input = self.TDNN(character_input)
File "
result = self.forward(*input, **kwargs)
File "~/pytorch_RVAE/selfModules/tdnn.py", line 42, in forward
xs = [x.max(2)[0].squeeze(2) for x in xs]
RuntimeError: dimension out of range (expected to be in range of [-2, 1], but got 2)
The text was updated successfully, but these errors were encountered: