pschwllr/MolecularTransformer

l cant understand the reason why i meet this error,can you give me some advice?Thanks

Closed this issue · 2 comments

batch_data = self.translate_batch(batch, data, fast=self.fast)

File "g:\pkg\MolecularTransformer1\onmt\translate\translator.py", line 344, in translate_batch
return self._translate_batch(batch, data)
File "g:\pkg\MolecularTransformer1\onmt\translate\translator.py", line 687, in _translate_batch
dec_states.beam_update(j, b.get_current_origin(), beam_size)
File "g:\pkg\MolecularTransformer1\onmt\decoders\decoder.py", line 408, in beam_update
sent_states.data.index_select(1, positions))
RuntimeError: index_select(): Expected dtype int32 or int64 for index

It could be related to the Pytorch version that you have installed (see: joeynmt/joeynmt#118).

Have you installed the packages as described in the readme?

Likely to be an issue related to the Pytorch version. Closing for now.