Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/train/
Damp.py200 update_cell = loss_scale_manager.get_update_cell()
201 if update_cell is not None:
209 scale_sense=update_cell).set_train()
212 … scale_sense=update_cell).set_train()
215 scale_sense=update_cell).set_train()
/third_party/mindspore/tests/ut/python/optimizer/
Dtest_optimizer_with_loss_scale.py220 update_cell = scale_manager.get_update_cell()
221 train_network = TrainOneStepWithLossScaleCell(net_with_loss, optimizer, scale_sense=update_cell)
237 update_cell = scale_manager.get_update_cell()
238 train_network = TrainOneStepWithLossScaleCell(net_with_loss, optimizer, scale_sense=update_cell)
254 update_cell = scale_manager.get_update_cell()
255 train_network = TrainOneStepWithLossScaleCell(net_with_loss, optimizer, scale_sense=update_cell)
271 update_cell = scale_manager.get_update_cell()
272 train_network = TrainOneStepWithLossScaleCell(net_with_loss, optimizer, scale_sense=update_cell)
Dtest_debug_location.py179 update_cell = scale_manager.get_update_cell()
180 train_network = TrainOneStepWithLossScaleCell(net_with_loss, optimizer, scale_sense=update_cell)
/third_party/mindspore/tests/perf_test/bert/
Dtest_bert_train.py180 update_cell = LossScaleUpdateCell(manager)
182 scale_update_cell=update_cell)
207 update_cell = LossScaleUpdateCell(manager)
209 scale_update_cell=update_cell)
/third_party/mindspore/tests/ut/python/parallel/
Dtest_loss_scale.py45 update_cell = DynamicLossScaleUpdateCell(loss_scale_value=65536, scale_factor=2, scale_window=1000) variable
188 net = TrainOneStepWithLossScaleCell(net, opt, update_cell)
201 net = nn.TrainOneStepWithLossScaleCell(net, opt, update_cell)
Dtest_parallel_transformer.py281 … update_cell = DynamicLossScaleUpdateCell(loss_scale_value=1024, scale_factor=2, scale_window=1000)
283 scale_sense=update_cell)
/third_party/mindspore/tests/st/model_zoo_tests/transformer/
Dtest_transformer.py179 update_cell = scale_manager.get_update_cell()
181 scale_update_cell=update_cell)