Model Not saving

[swetha-sree]

The main_train.py was successfully initialized it was able to create folder structure as well, but once the process ends there is no '.pt' model begin saved.

Can you please help on what i should be doing in order to save a model.

[mahdidrm]

The main_train.py was successfully initialized it was able to create folder structure as well, but once the process ends there is no '.pt' model begin saved.

Can you please help on what i should be doing in order to save a model.

Hi,
I have same problem, I changed the 500 iteration to 300 and the have saved .But in test step, I have no results and S3 folders in Output are epmty!

Please check it and tell me if it works for you.