前提
pytorch_lightningで@pl.data_loderにエラーがでます。
pytorch_lightningのVerは1.7.7です。
解決策をお待ちしております。
実現したいこと
エラーの解消
発生している問題・エラーメッセージ
AttributeError Traceback (most recent call last) Input In [24], in <cell line: 1>() ----> 1 class Net(pl.LightningModule): 2 # New: バッチサイズ等を引数に指定 3 def __int__(self, input_size=10, hidden_size=5, output_size=7, batch_size=10): 4 super(Net, self).__init__() Input In [24], in Net() 22 return torch.optim.SGD(self.parameters(),lr=0.1) 24 # New:train用のDataLoderの設定 ---> 25 @pl.data_loder 26 def train_dataloder(self): 27 return torch.utils.data.DataLoder(train, self.batch_size, shuffle=True) 29 # New:学習データに対する処理 AttributeError: module 'pytorch_lightning' has no attribute 'data_loder'
該当のソースコード
python
1class Net(pl.LightningModule): 2# New: バッチサイズ等を引数に指定 3 def __int__(self, input_size=10, hidden_size=5, output_size=7, batch_size=10): 4 super(Net, self).__init__() 5 self.fc1 = nn.Linear(input_size, hidden_size) 6 self.fc2 = nn.Linear(hidden_size, output_size) 7 self.batch_size = batch_size 8 9# 変更なし 10 def forward(self, x): 11 x = self.fc1(x) 12 x = F.relu(x) 13 x = self.fc2(x) 14 return x 15 16# New:目的変数の設定 lossfunは目的関数 17 def lossfun(self, y, t): 18 return F.cross_entropy(y,t) 19 20# New:optimizerの設定 configure optimizerは最適化手法の設定 21 def configure_optimizers(self): 22 return torch.optim.SGD(self.parameters(),lr=0.1) 23 24# New:train用のDataLoderの設定 25 @pl.data_loder 26 def train_dataloder(self): 27 return torch.utils.data.DataLoder(train, self.batch_size, shuffle=True) 28 29# New:学習データに対する処理 30 def training_step(self, batch, batch_nb): 31 x, t = batch 32 y = self.forword(x) 33 loss = self.lossfun(y, t) 34 results = {'loss':loss} 35 return results
試したこと
補足情報(FW/ツールのバージョンなど)
ここにより詳細な情報を記載してください。

バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2022/10/11 01:32
2022/10/11 01:43 編集