Searched defs:train_network (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | test_multi_full_ps.py | 106 train_network = TrainOneStepCell(net_with_criterion, net_opt) variable
|
| /third_party/mindspore/tests/st/probability/transforms/ |
| D | test_transform_bnn_layer.py | 132 train_network = TrainOneStepCell(net_with_loss, optimizer) variable
|
| D | test_transform_bnn_model.py | 131 train_network = TrainOneStepCell(net_with_loss, optimizer) variable
|
| /third_party/mindspore/tests/st/fl/cross_silo_lenet/ |
| D | test_cross_silo_lenet.py | 108 train_network = TrainOneStepCell(net_with_criterion, net_opt) variable
|
| /third_party/mindspore/tests/ut/python/train/summary/ |
| D | test_image_summary.py | 156 def record(self, step, train_network=None): argument
|
| /third_party/mindspore/tests/st/fl/mobile/ |
| D | test_mobile_lenet.py | 134 train_network = TrainOneStepCell(net_with_criterion, net_opt) variable
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/ |
| D | test_hybrid_train_lenet.py | 154 train_network = TrainOneStepCellWithServerCommunicator(net_with_criterion, net_opt) variable
|
| /third_party/mindspore/mindspore/train/summary/ |
| D | summary_record.py | 296 def record(self, step, train_network=None, plugin_filter=None): argument
|
| /third_party/mindspore/mindspore/train/callback/ |
| D | _summary_collector.py | 49 train_network = 'train_network' variable in LineageMetadata
|
| /third_party/mindspore/mindspore/train/ |
| D | model.py | 1092 def train_network(self): member in Model
|