From 54f4d38148689c2edffa8b82fa0f55871961d0fb Mon Sep 17 00:00:00 2001
From: leo <yuwl798180@163.com>
Date: Fri, 6 Dec 2019 10:44:20 +0800
Subject: [PATCH] pretty logger

---
 main.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 8959c16..0935d73 100644
--- a/main.py
+++ b/main.py
@@ -23,6 +23,7 @@ def main(cfg):
     cwd = utils.get_original_cwd()
     cfg.cwd = cwd
     cfg.pos_size = 2 * cfg.pos_limit + 2
+    logger.info(f'\n{cfg.pretty()}')
 
     __Model__ = {
         'cnn': models.PCNN,
@@ -66,8 +67,6 @@ def main(cfg):
 
     model = __Model__[cfg.model_name](cfg)
     model.to(device)
-
-    logger.info(f'\n{cfg.pretty()}')
     logger.info(f'\n {model}')
 
     optimizer = optim.Adam(model.parameters(), lr=cfg.learning_rate, weight_decay=cfg.weight_decay)