Home
last modified time | relevance | path

Searched refs:loss_callback (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/st/model_zoo_tests/DeepFM/
Dtest_deepfm.py59 loss_callback = LossCallBack(loss_file_path=loss_file_name)
60 callback_list = [time_callback, loss_callback]
75 print("loss_callback.loss:", loss_callback.loss)
76 assert loss_callback.loss < export_loss_value
/third_party/mindspore/tests/st/networks/models/bert/bert_performance/
Dtest_bert_thor.py174 loss_callback = LossCallback()
175 callback = [time_monitor_callback, loss_callback]
186 loss_list = loss_callback.loss_list
/third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/
Dconfig.py62 loss_callback = True variable in TrainConfig
Ddeepfm.py383 if self.train_config.loss_callback:
384 loss_callback = LossCallBack(loss_file_path=os.path.join(self.train_config.output_path,
386 callback_list.append(loss_callback)