Create memo.md

init
msakasai · Mar 24, 2020 · f2d866a · f2d866a
1 parent 98db04b
commit f2d866a
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/MachineLearning/DeepLearning/PyTorch/memo.md b/MachineLearning/DeepLearning/PyTorch/memo.md
@@ -0,0 +1,11 @@
+## モデルの保存
+```python
+torch.save(model.state_dict(), save_path) # 学習済みモデルパラメータ, 保存先パス
+```
+
+## モデルの読み込み
+```python
+model = ModelClass(*args, **kwargs)
+model.load_state_dict(torch.load(save_path))
+```
+