!8535 add export air test

From: @changzherui
Reviewed-by: @kingxian
Signed-off-by: @kingxian
pull/8535/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 929d31653d

@ -36,7 +36,7 @@ class TimeMonitor(Callback):
self.epoch_time = time.time()
def epoch_end(self, run_context):
epoch_seconds = (time.time() - self.epoch_time) * 1000
epoch_seconds = time.time() - self.epoch_time
step_size = self.data_size
cb_params = run_context.original_args()
if hasattr(cb_params, "batch_num"):
@ -49,4 +49,4 @@ class TimeMonitor(Callback):
return
step_seconds = epoch_seconds / step_size
print("Epoch time: {:5.3f}, per step time: {:5.3f}".format(epoch_seconds, step_seconds), flush=True)
print("Epoch time: {:5.3f}s, per step time: {:5.3f}s".format(epoch_seconds, step_seconds), flush=True)

@ -378,7 +378,8 @@ def load_param_into_net(net, parameter_dict, strict_load=False):
logger.debug("%s", param_name)
logger.info("Load parameter into net finish.")
logger.warning("{} parameters in the net are not loaded.".format(len(param_not_load)))
if param_not_load:
logger.warning("{} parameters in the net are not loaded.".format(len(param_not_load)))
return param_not_load

File diff suppressed because it is too large Load Diff
Loading…
Cancel
Save