THUMT icon indicating copy to clipboard operation
THUMT copied to clipboard

pytorch version ? Providing a bool or integral fill value without setting the optional `dtype` or `out` arguments is currently unsupported. In PyTorch 1.7,

Open anbo724 opened this issue 3 years ago • 2 comments

训练报错 Providing a bool or integral fill value without setting the optional dtype or out arguments is currently unsupported. In PyTorch 1.7, 什么版本的pytoch可以使用?

anbo724 avatar Jun 18 '21 08:06 anbo724

训练报错 Providing a bool or integral fill value without setting the optional dtype or out arguments is currently unsupported. In PyTorch 1.7, 什么版本的pytoch可以使用?

可以考虑使用低版本的PyTorch,比如1.4版本,如果方便的话,请告知一下具体报错的代码位置,我们会尽快修复这个bug,谢谢!

GrittyChen avatar Jun 18 '21 08:06 GrittyChen

训练报错 Providing a bool or integral fill value without setting the optional dtype or out arguments is currently unsupported. In PyTorch 1.7, 什么版本的pytoch可以使用?

可以考虑使用低版本的PyTorch,比如1.4版本,如果方便的话,请告知一下具体报错的代码位置,我们会尽快修复这个bug,谢谢!

在inference.py中的init_seqs = torch.full([batch_size, beam_size, 1], bos_id, device=device)加上dtype=torch.long应该可以解决。

leileilin avatar Nov 29 '21 07:11 leileilin