【pytorch】ValueError: optimizer got an empty parameter list　のエラーが発生

前提・実現したいこと

pytorchでLSTMを実行したい。

最適化関数を定義する際に以下のエラーメッセージが発生しました。

発生している問題・エラーメッセージ

ValueError: optimizer got an empty parameter list

該当のソースコード

python
1class LSTM(nn.Module):
2
3    #クラスのインスタンス化（はじめにLSTMクラスが実行されたときのみ実行される)
4    def __init__(self, input_size =2, hidden_size =50, output_size = 1 ):
5        super().__init__()
6
7        #隠れ層のサイズ
8        self.hidden_size = hidden_size,
9
10        #input_size　→　hidden_sizeに変換
11        self.lstm = nn.LSTM(input_size, hidden_size),
12
13        #hidden_size →　output_sizeに変換　lstmの出力に全結合レイヤーを介す
14        self.fc = nn.Linear(hidden_size, output_size),
15
16        #0で隠れ層を初期化する、hidden層の初期化とcell層の初期化
17        self.hidden = (torch.zeros(2,1,hidden_size),torch.zeros(2,1,hidden_size))
18
19
20    #予測に用いる関数(順伝播)
21    def forward(self, x):
22        #データと隠れ層の状態が返ってくる
23        lstm_out , self.hidden = self.lstm(x, self.hidden)
24        #全結合層、順伝播させるのはデータが格納されているlstm_outのみ
25        pred = self.fc(lstm_out)
26
27        return pred[-1] #-1を指定することで一番最後の値を取り出すことができる
28
29
30print()
31print('********************LSTMの定義終了********************')
32print()
33
34
35''' ******************関数の定義・インスタンス化****************** '''
36
37#モデルのインスタンス化と損失関数、最適化関数を定義する
38torch.manual_seed(3) #乱数の値を一定にし、処理結果を同じにする
39model = LSTM() #インスタンス
40
41param = model.parameters()
42print(param)
43
44print('***********************************')
45
46criterion = nn.MSELoss() #損失関数
47optimizer = torch.optim.SGD(param,lr=0.01) #最適化関数
48