Comment by vivahir215

ah okay. Didnt know this. I generally use pytorch save models for my workflow.