nikkie-ftnextの日記

イベントレポートや読書メモを発信

2024-09-18から1日間の記事一覧

PyTorchのモデル(nn.Module)には、これから訓練に入ることをtrain()メソッドで教えてあげよう

はじめに ぴえぴえ... nikkieです。 Today I learnedです。 開発者がtorch.nn.Moduleのお世話をするんだ! 目次 はじめに 目次 私にはTrainerの中がブラックボックス nn.Module.train()やnn.Module.eval()ってなんだ? モデル(nn.Module)には、modeがある …