训练一个epoch后模型很大的问题

[IsDora]

您好，想问一下，为什么训练一个epoch后得到的模型很大呢，官方的显示只有46M。需要有什么改动吗

[IsDora]

每个epoch训练出来的模型大小都不一样，你们有遇到这个问题吗

[atomishcv]

@IsDora 请问你训练时出现过loss值为nan的情况吗？

[guanfang12]

Because this code use the .double() data type, and it use torch.save(model, model_out_path) rather than torch.save(state_dict, model_out_path). If you change it to float(), it will be 46M.