THUMT
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,
训练报错 Providing a bool or integral fill value without setting the optional dtype
or out
arguments is currently unsupported. In PyTorch 1.7,
什么版本的pytoch可以使用?
训练报错 Providing a bool or integral fill value without setting the optional
dtype
orout
arguments is currently unsupported. In PyTorch 1.7, 什么版本的pytoch可以使用?
可以考虑使用低版本的PyTorch,比如1.4版本,如果方便的话,请告知一下具体报错的代码位置,我们会尽快修复这个bug,谢谢!
训练报错 Providing a bool or integral fill value without setting the optional
dtype
orout
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应该可以解决。