2021-12-25から1日間の記事一覧
medium.com
askubuntu.com sudo apt install renameutils EDITOR="vim" qmv -f do
tzmi.hatenablog.comGPUで保存→GPUで呼び出す場合 保存 model_path = 'model.pth' torch.save(model.state_dict(), model_path)呼び出し model_path = 'model.pth' model.load_state_dict(torch.load(model_path)) GPUで学習→CPUで保存 model_path = 'model.…
ここを参考に。 qiita.com 学習済みモデルにはResNet-18を使用。 データセットは以下を使用。 Potato Leaf Disease Datasetwww.kaggle.com Train: 3ラベル(Healthy, Early blight, Late blight) 各300枚 Validation: 3ラベル 各100枚実行結果 Training compl…