Programming/Python
2024. 11. 14.
[torch] 모델 저장하고 경고문없이 로드하기
1. 모델 저장하기 import copybest_model = copy.deepcopy(model.state_dict())torch.save(best_model, 'best.pth') 2. 모델 로드하기 단 여기서 model param이 앞에 정의되어있어야 함model.load_state_dict(torch.load('best.pth')) 이렇게 로드하면 FutureWarning: You are using `torch.load` with `weights_only=False` (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data..