Hey! Trains Looks Super Promising For Our Project, But I Struggle To Understand The Saving And Model Aspect Of Trains. I Have A Pytorch Model That I Save With Some Additional Data Every Epoch And I Would Like To Integrate That With Trains. Is That Possibl

Unanswered

Trains do patch the torch save function 🙂

If you like, you can save the model for each epoch by having a unique name for it. The model will be saved in the output_uri path you have in the Task.init command.
For example, this code will save a model for every epoch:

for epoch in range(num_of_epoch): # Create a model torch.save(model, "model_epoch_{}".format(epoch))

  				
Posted 
	5 years ago

					More
				  		
  Report
		
					TimelyPenguin76
				
					0
					 Administrator

390 Views

0 Answers

5 years ago

2 years ago