Searched defs:amp_level (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/train/train_thor/ |
| D | convert_utils.py | 181 …def convert_to_thor_model(model, network, loss_fn=None, optimizer=None, metrics=None, amp_level="O… argument
|
| D | model_thor.py | 112 eval_indexes=None, amp_level="O0", **kwargs): argument
|
| /third_party/mindspore/mindspore/train/ |
| D | model.py | 135 eval_indexes=None, amp_level="O0", boost_level="O0", **kwargs): argument
182 def _check_amp_level_arg(self, optimizer, amp_level): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | model_thor.py | 139 eval_indexes=None, amp_level="O0", frequency=278, stop_epoch=100, **kwargs): argument
|