Searched refs:loss_cb (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/tests/ut/python/utils/ |
| D | test_callback.py | 127 loss_cb = LossMonitor(1)
128 callbacks = [loss_cb]
142 loss_cb = LossMonitor(1)
148 loss_cb.begin(run_context)
149 loss_cb.epoch_begin(run_context)
150 loss_cb.step_begin(run_context)
151 loss_cb.step_end(run_context)
152 loss_cb.epoch_end(run_context)
153 loss_cb.end(run_context)
|
| /third_party/mindspore/tests/st/networks/models/resnet50/ |
| D | test_resnet50_imagenet.py | 209 loss_cb = LossGet(1, step_size)
216 model.train(1, dataset, callbacks=loss_cb)
221 time_cost = loss_cb.get_per_step_time()
222 loss = loss_cb.get_loss()
289 loss_cb = LossGet(1, step_size)
296 model.train(eval_interval, dataset, callbacks=loss_cb)
301 time_cost = loss_cb.get_per_step_time()
302 loss = loss_cb.get_loss()
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | test_resnet_cifar_8p.py | 169 loss_cb = LossGet()
170 model.train(epoch_size, dataset, callbacks=[loss_cb])
171 q.put(loss_cb.get_loss())
|
| D | test_resnet_cifar_1p.py | 144 loss_cb = LossGet()
145 model.train(epoch_size, dataset, callbacks=[loss_cb])
|
| D | resnet_cifar.py | 144 loss_cb = LossMonitor() variable
145 model.train(epoch_size, dataset, callbacks=[ckpoint_cb, loss_cb, time_cb])
|
| /third_party/mindspore/tests/st/mem_reuse/ |
| D | resnet_cifar_memreuse.py | 142 loss_cb = LossMonitor() variable
143 model.train(epoch_size, dataset, callbacks=[ckpoint_cb, loss_cb])
|
| D | resnet_cifar_normal.py | 142 loss_cb = LossMonitor() variable
143 model.train(epoch_size, dataset, callbacks=[ckpoint_cb, loss_cb])
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/ |
| D | test_fl_fasterrcnn.py | 196 loss_cb = LossCallBack(rank_id=rank)
197 cb = [time_cb, loss_cb]
|
| /third_party/mindspore/tests/st/fl/cross_silo_femnist/ |
| D | test_cross_silo_femnist.py | 321 loss_cb = LossGet(1, num_batches)
323 cbs.append(loss_cb)
346 loss_list = loss_cb.get_loss()
|