Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!48418 fix dataset_sink_mode of DeepFM 

Merge pull request !48418 from zhangfanghe/fix_deepFM_bug
This commit is contained in:
i-robot 2023-02-06 06:23:44 +00:00 committed by Gitee
parent f33b9f7258 628c1a2c16
commit cef9248dd8
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/model_zoo_tests/DeepFM/test_deepfm.py
View File

@ -69,7 +69,7 @@ def test_deepfm():
callback_list.append(eval_callback)
print("train_config.train_epochs:", train_config.train_epochs)
model.train(train_config.train_epochs, ds_train, callbacks=callback_list)
model.train(train_config.train_epochs, ds_train, callbacks=callback_list, dataset_sink_mode=True)
export_loss_value = 0.52
print("loss_callback.loss:", loss_callback.loss)