imporve msg info
@@ -58,7 +58,6 @@ def main(args):
if use_gpu:
gpu_id = ParallelEnv().dev_id
place = paddle.CUDAPlace(gpu_id)
- print("[gry debug]gpu_id: ", gpu_id)
else:
place = paddle.CPUPlace()
@@ -85,6 +84,7 @@ def main(args):
valid_dataloader = Reader(config, 'valid', places=place)()
best_top1_acc = 0.0 # best top1 acc record
+ best_top1_epoch = 0
for epoch_id in range(config.epochs):
net.train()
# 1. train with train dataset
@@ -99,14 +99,14 @@ def main(args):
None, epoch_id, 'valid')
if top1_acc > best_top1_acc:
best_top1_acc = top1_acc
- message = "The best top1 acc {:.5f}, in epoch: {:d}".format(
- best_top1_acc, epoch_id)
- logger.info("{:s}".format(logger.coloring(message, "RED")))
+ best_top1_epoch = epoch_id
if epoch_id % config.save_interval == 0:
-
model_path = os.path.join(config.model_save_dir,
config.ARCHITECTURE["name"])
save_model(net, optimizer, model_path, "best_model")
+ message = "The best top1 acc {:.5f}, in epoch: {:d}".format(
+ best_top1_acc, epoch_id)
+ logger.info("{:s}".format(logger.coloring(message, "RED")))
# 3. save the persistable model
if epoch_id % config.save_interval == 0:
