pyTorch2 PyTorch checkpoint # Load model model = MyModel().cuda() model_data = torch.load('model_path') model.load_state_dict(model_data['model_state_dict'], strict=False) optimizer = optim.Adam(model.parameters()) optimizer.load_state_dict(model_data['optimizer_state_dict']) # Save model torch.save({ 'epoch': epoch, 'model_state_dict': model.state_dict(), 'optimizer_state_dict': optimizer.state_dict() }, os.path.join('sav.. 2021. 12. 15. [PyTorch] ModuleNotFoundError: No module named 'dateutil' ModuleNotFoundError: No module named 'dateutil' pip install python-dateutil (pip3 install python-dateutil) 2021. 2. 1. 이전 1 다음