note engineers’ notes

140

PyTorch Moduleに紐づく定数のtensorを定義する

環境python 3.6.6IPython 7.1.1torch==1.7.0 目的PyTorchで`torch.nn.Module`に紐づく定数を定義したい場合があるとします.`torch.nn.parameter.Parameter`で定義せず`torch.tensor`に`require_grad=False`にすれば,`optimizer`のstepメソッドでは学習はされませんが,toメソッドでGPUに乗せたり,data typeを変えることができません. 解決方

スキ
4