shenweichen / DeepCTR

Easy-to-use,Modular and Extendible package of deep-learning based CTR models .
https://deepctr-doc.readthedocs.io/en/latest/index.html
Apache License 2.0
7.54k stars 2.21k forks source link

如何保存deepctr-torch训练好的deepfm模型 #527

Closed RainFung closed 1 year ago

RainFung commented 1 year ago

Please refer to the FAQ in doc and search for the related issues before you ask the question.

Describe the question(问题描述)

image
RainFung commented 1 year ago

thanks

LEI CHEN @.***> 于2023年8月1日周二 13:11写道:

you can save in the script "base_model.py" by using torch.save(model.state_dict(), './data/save_dict/deep_fmepoch{}.ckpt'.format(epoch)) at last in fit function

— Reply to this email directly, view it on GitHub https://github.com/shenweichen/DeepCTR/issues/527#issuecomment-1659578185, or unsubscribe https://github.com/notifications/unsubscribe-auth/AFQM5A42TJIEIWLWWCDUDQDXTCFZHANCNFSM6AAAAAA26E6WYY . You are receiving this because you authored the thread.Message ID: @.***>