Searched defs:train_wrap (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/examples/transfer_learning/model/ |
| D | train_utils.py | 20 def train_wrap(net, loss_fn=None, optimizer=None, weights=None): function
|
| /third_party/mindspore/mindspore/lite/examples/train_lenet_java/model/ |
| D | train_utils.py | 21 def train_wrap(net, loss_fn=None, optimizer=None, weights=None): function
|
| /third_party/mindspore/mindspore/lite/examples/unified_api/model/ |
| D | train_utils.py | 21 def train_wrap(net, loss_fn=None, optimizer=None, weights=None): function
|
| /third_party/mindspore/mindspore/lite/examples/train_lenet/model/ |
| D | train_utils.py | 22 def train_wrap(net, loss_fn=None, optimizer=None, weights=None, loss_scale_manager=None): function
|
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | train_utils.py | 22 def train_wrap(net, loss_fn=None, optimizer=None, weights=None): function
|