Sure Simple MNIST train toy example
@<1523701070390366208:profile|CostlyOstrich36>
if I use this run train without fire cli:

if __name__ == '__main__':
     main()

@<1523701087100473344:profile|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 @<1577830989026037760:profile|EnormousGiraffe79> , can you please a self contained code snippet that reproduces this?

Hi @<1577830989026037760:profile|EnormousGiraffe79> , are you seeing any errors?