Как преобразовать параметры модели Pytorch в длинный тип данных? - PullRequest
0 голосов
/ 07 июля 2019

Мне удалось преобразовать параметры модели Pytorch во всплывающее или двойное число, но не в длинное.

model = model.long() 

выдает ошибку, в то время как

model = model.float() 

работает.

Я получаю ошибку:

'Net' object has no attribute 'long'

1 Ответ

2 голосов
/ 07 июля 2019

Большинство nn модулей не поддерживают длинные (целочисленные) операции, например свертки, линейный слой и т. Д. Следовательно, вы не можете "привести" модель к torch.long.

...