Searched refs:LossCallBack (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | train_and_test_multinpu_ci.py | 27 from src.callbacks import LossCallBack, EvalCallBack
61 callback = LossCallBack()
97 callback = LossCallBack(config=config)
|
| D | callbacks.py | 31 class LossCallBack(Callback): class
45 super(LossCallBack, self).__init__()
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/ |
| D | train_and_test_multinpu_ci_data_parallel.py | 26 from src.callbacks import LossCallBack, EvalCallBack
58 callback = LossCallBack()
95 callback = LossCallBack(config=config)
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/ |
| D | test_deeplabv3.py | 39 class LossCallBack(Callback): class
48 super(LossCallBack, self).__init__()
86 callback = LossCallBack(dataset_size)
|
| /third_party/mindspore/tests/st/model_zoo_tests/deeplabv3/ |
| D | train_one_epoch_with_loss.py | 35 class LossCallBack(Callback): class
44 super(LossCallBack, self).__init__()
79 callback = LossCallBack(dataset_size)
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/ |
| D | test_deepfm.py | 26 from src.callback import EvalCallBack, LossCallBack, TimeMonitor
59 loss_callback = LossCallBack(loss_file_path=loss_file_name)
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | callback.py | 53 class LossCallBack(Callback): class
64 super(LossCallBack, self).__init__()
|
| D | deepfm.py | 34 from .callback import EvalCallBack, LossCallBack
384 loss_callback = LossCallBack(loss_file_path=os.path.join(self.train_config.output_path,
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/ |
| D | network_define.py | 31 class LossCallBack(Callback): class
45 super(LossCallBack, self).__init__()
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/ |
| D | test_fl_fasterrcnn.py | 33 from src.network_define import LossCallBack, WithLossCell, TrainOneStepCell, LossNet
196 loss_cb = LossCallBack(rank_id=rank)
|