如何将Pytorch模型参数转换为long数据类型？

Question

我能够将pytorch模型参数转换为float或double，但不能转换为long。

model = model.long() 

给出错误，而

model = model.float() 

运行。

我得到的错误是：

'Net' object has no attribute 'long'

Answer 1

大多数nn模块不支持长（整数）运算，例如卷积，线性层等。因此，您无法将模型“投射”到torch.long。