如何保存deepctr-torch训练好的deepfm模型
Closed this issue · 1 comments
RainFung commented
Please refer to the FAQ in doc and search for the related issues before you ask the question.
Describe the question(问题描述)
RainFung commented
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_fm_epoch_{}.ckpt'.format(epoch)) at last in fit
function
—
Reply to this email directly, view it on GitHub
<#527 (comment)>,
or unsubscribe
<https://github.com/notifications/unsubscribe-auth/AFQM5A42TJIEIWLWWCDUDQDXTCFZHANCNFSM6AAAAAA26E6WYY>
.
You are receiving this because you authored the thread.Message ID:
***@***.***>