Hi EnormousGiraffe79 , are you seeing any errors?

SuccessfulKoala55
No, I didn't see any errors. When I save model
torch.save(model.state_dict(), os.path.join(gettempdir(), " mny_model.pt "))
clearML task don't catch it when I run script from Fire

Hi EnormousGiraffe79 , can you please a self contained code snippet that reproduces this?

Sure Simple MNIST train toy example
CostlyOstrich36
if I use this run train without fire cli:

if __name__ == '__main__':
     main()