From 28c81813c66b8f29045221c9c5d7d6c4e08436f6 Mon Sep 17 00:00:00 2001 From: yepei6 Date: Thu, 10 Sep 2020 20:35:16 +0800 Subject: [PATCH] update model_zoo/official/cv/mobilenetv2/train.py. delete the comment of printing training info --- model_zoo/official/cv/mobilenetv2/train.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/model_zoo/official/cv/mobilenetv2/train.py b/model_zoo/official/cv/mobilenetv2/train.py index 64e58250ef..151ac441bf 100644 --- a/model_zoo/official/cv/mobilenetv2/train.py +++ b/model_zoo/official/cv/mobilenetv2/train.py @@ -108,10 +108,6 @@ if __name__ == '__main__': losses.append(network(feature, label).asnumpy()) epoch_mseconds = (time.time()-epoch_start) * 1000 per_step_mseconds = epoch_mseconds / step_size - # lr cause to pynative, but cpu doesn't support this mode - # print("\r epoch[{}], iter[{}] cost: {:5.3f}, per step time: {:5.3f}, avg loss: {:5.3f}, lr: {}"\ - # .format(epoch + 1, step_step, epoch_mseconds, per_step_mseconds, np.mean(np.array(losses)), \ - # lr[(epoch+1)*step_size - 1]), end="") print("\r epoch[{}], iter[{}] cost: {:5.3f}, per step time: {:5.3f}, avg loss: {:5.3f}"\ .format(epoch + 1, step_size, epoch_mseconds, per_step_mseconds, np.mean(np.array(losses))), \ end="")