search

danfenghong / IEEE_TGRS_SpectralFormer

Danfeng Hong, Zhu Han, Jing Yao, Lianru Gao, Bing Zhang, Antonio Plaza, Jocelyn Chanussot. Spectralformer: Rethinking hyperspectral image classification with transformers, IEEE Transactions on Geoscience and Remote Sensing (TGRS), 2021
219 stars 39 forks source link

save the training model? #8

Open volare1996 opened 2 years ago

volare1996 commented 2 years ago

Hong老师您好： 请问下我在训练完成后如何保存模型呢？即您所给的log里模型！ 谢谢

danfenghong commented 2 years ago

你好，感谢对我们工作的兴趣。有专门的函数，torch.save， 可以保存成.pth文件。 祝好！ 丹枫

volare1996 @.***> 于2022年1月7日周五 11:00写道：

Hong老师您好： 请问下我在训练完成后如何保存模型呢？即您所给的log里模型！ 谢谢

— Reply to this email directly, view it on GitHub https://github.com/danfenghong/IEEE_TGRS_SpectralFormer/issues/8, or unsubscribe https://github.com/notifications/unsubscribe-auth/AFL2GZQ7P6F6PCX5AQKV64LUUZJMFANCNFSM5LN25BJA . Triage notifications on the go with GitHub Mobile for iOS https://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675 or Android https://play.google.com/store/apps/details?id=com.github.android&referrer=utm_campaign%3Dnotification-email%26utm_medium%3Demail%26utm_source%3Dgithub.

You are receiving this because you are subscribed to this thread.Message ID: @.***>