Home
last modified time | relevance | path

Searched refs:ModelCallback (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/tests/st/auto_parallel/
Dsoft_entropy_loss_expand_parallel.py124 class ModelCallback(Callback): class
126 super(ModelCallback, self).__init__()
194 single_callback = ModelCallback()
205 parallel_callback = ModelCallback()
217 parallel_callback = ModelCallback()
243 parallel_callback = ModelCallback()
Dresnet50_expand_loss.py351 class ModelCallback(Callback): class
353 super(ModelCallback, self).__init__()
364 parallel_callback = ModelCallback()
/third_party/mindspore/tests/st/model_zoo_tests/yolov3/
Dtest_yolov3.py60 class ModelCallback(Callback): class
62 super(ModelCallback, self).__init__()
131 model_callback = ModelCallback()
/third_party/mindspore/tests/st/model_zoo_tests/transformer/
Dtest_transformer.py115 class ModelCallback(Callback): class
117 super(ModelCallback, self).__init__()
174 callback = ModelCallback()
/third_party/mindspore/tests/st/networks/models/bert/bert_precision/
Dtest_bert_tdt_lossscale.py144 class ModelCallback(Callback): class
146 super(ModelCallback, self).__init__()
203 callback = ModelCallback()
/third_party/mindspore/tests/st/networks/models/bert/bert_performance/
Dtest_bert_tdt_lossscale.py144 class ModelCallback(Callback): class
146 super(ModelCallback, self).__init__()
207 callback = ModelCallback()
/third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/
Dtest_yolov3_darknet53.py63 class ModelCallback(Callback): class
65 super(ModelCallback, self).__init__()