Open ramyaragh opened 6 years ago
delete .squeeze(2)
Does not seem to help
preprocessed data was found and loaded
Traceback (most recent call last):
File "train.py", line 59, in
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
z=None)
File "~/pytorch/torch/lib/python2.7/site-packages/torch/nn/modules/module.py", line 325, in call
result = self.forward(*input, kwargs)
File "~/pytorch_RVAE/model/rvae.py", line 64, in forward
encoder_input = self.embedding(encoder_word_input, encoder_character_input)
File "/~/pytorch/torch/lib/python2.7/site-packages/torch/nn/modules/module.py", line 325, in call
result = self.forward(*input, *kwargs)
File "~/pytorch_RVAE/selfModules/embedding.py", line 47, in forward
character_input = self.TDNN(character_input)
File "~/pytorch/torch/lib/python2.7/site-packages/torch/nn/modules/module.py", line 325, in call
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)