You can call torch.save() multiple times in your training routine and save the model’s data in different output files.
torch.save()