[キャンプAI技術を導く]2週目Day 3
4153 ワード
今日の勉強内容
1. Load & Save Model
2. Monitoring tools for PyTorch
1. Load & Save Model
学習モデルには多くの時間がかかり,学習中にWeightなどを格納する必要がある.
model.save()
を使用すると、モデルのすべての形状を保存するか、パラメータのみを保存できます.model.state_dict
には、モデルのパラメータ情報が存在する.import torch
...
...
# 모델의 파라미터를 Save
torch.save(model.state_dict(), os.path.join(PATH, "model.pt")
# 동일한 모델의 형태에서 파라미터만을 Load
new_model = Model()
new_model.load_statae_dict(torch.load(os.path.join(PATH, "model.pt")))
# 모델의 구조와 파라미터를 모두 Save
torch.save(model, os.path.join(PATH, "model.pt")
# 모델의 구조와 파라미터를 모두 Load
new_model2 = torch.load(os.path.join(PATH, "model.pt")
学習中に結果を保存して最適なモデルを最終的に選択
測定値を継続的にチェックして格納し、一定数のAccまたはAccが改善されていない場合に学習を停止→自動的に学習を停止する
Transfer Learning
2. Monitoring tools for PyTorch
TensorBoard VS Weight & Biases(WandB)
ソース:https://www.tensorflow.org/tensorboard?hl=ko
ソース:https://wandb.ai/site
これは
Reference
この問題について([キャンプAI技術を導く]2週目Day 3), 我々は、より多くの情報をここで見つけました https://velog.io/@mimsu1139/부스트캠프-AI-Tech-2주차-Day-3テキストは自由に共有またはコピーできます。ただし、このドキュメントのURLは参考URLとして残しておいてください。
Collection and Share based on the CC Protocol