Append model checkpoints to existing file in PyTorch
Append model checkpoints to existing file in PyTorch Question: In PyTorch, it is possible to save model checkpoints as follows: import torch # Create a model model = torch.nn.Sequential( torch.nn.Linear(1, 50), torch.nn.Tanh(), torch.nn.Linear(50, 1) ) # … some training here # Save checkpoint torch.save(network.state_dict(), ‘checkpoint.pt’) During my training procedure, I save a checkpoint every 100 …
Total answers: 1