쉬엄쉬엄블로그

이 색깔은 주석이라 무시하셔도 됩니다. 학습 결과를 공유하고 싶다면? 학습 결과를 저장할 필요가 있다. model.save() 학습의 결과를 저장하기 위한 함수 모델 형태(architecture)와 파라미터를 저장 파라미터만 저장하면 적은 용량으로 저장 가능 모델 학습 중간 과정의 저장을 통해 최선의 결과모델을 선택 만들어진 모델을 외부 연구자와 공유하여 학습 재연성 향상 import torch import torch.nn as nn import torch.optim as optim import os class TheModelClass(nn.Module): def __init__(self): super(TheModelClass, self).__init__() self.layer1 = nn.Sequenti..