Unanswered
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
Hi UnsightlyShark53 ,
If you want to save anything from your experiment, you can use the artifacts, you can view this example - https://github.com/allegroai/trains/blob/master/examples/artifacts_toy.py
Can this do the trick?
229 Views
0
Answers
4 years ago
one year ago