Searched refs:TrainStepWrap (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/ |
| D | train_step_wrap.py | 23 class TrainStepWrap(nn.Cell): class
29 super(TrainStepWrap, self).__init__()
59 return TrainStepWrap(NetWithLossClass(network))
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_momentum.py | 77 class TrainStepWrap(nn.Cell): class
81 super(TrainStepWrap, self).__init__()
122 'block': TrainStepWrap(NetWithLossClass(Net())),
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_semi_auto_two_subgraphs.py | 67 class TrainStepWrap(nn.Cell): class
69 super(TrainStepWrap, self).__init__()
107 net = TrainStepWrap(NetWithLoss(Net()))
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | train_and_test_multinpu_ci.py | 26 from src.wide_and_deep import PredictWithSigmoid, TrainStepWrap, NetWithLossClass, WideDeepModel
43 train_net = TrainStepWrap(loss_net)
|
| D | wide_and_deep.py | 253 class TrainStepWrap(nn.Cell): class
264 super(TrainStepWrap, self).__init__()
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/ |
| D | train_and_test_multinpu_ci_data_parallel.py | 25 from src.wide_and_deep import PredictWithSigmoid, TrainStepWrap, NetWithLossClass, WideDeepModel
40 train_net = TrainStepWrap(loss_net)
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | deepfm.py | 292 class TrainStepWrap(nn.Cell): class
297 super(TrainStepWrap, self).__init__(auto_prefix=False)
394 train_net = TrainStepWrap(loss_net, lr=self.train_config.learning_rate,
|