From 5058d873501543ad79387dcaafe11033bd8c4c92 Mon Sep 17 00:00:00 2001
From: fnakamura <fujiki.nakamura@gmail.com>
Date: Wed, 20 Feb 2019 22:23:07 +0900
Subject: [PATCH] add logger

---
 train_source.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/train_source.py b/train_source.py
index edcd74c..0481722 100644
--- a/train_source.py
+++ b/train_source.py
@@ -8,11 +8,14 @@ from torchvision import transforms
 
 from models import CNN
 from trainer import train_source_cnn
+from utils import get_logger
 
 
 def main(args):
     if not os.path.exists(args.logdir):
         os.makedirs(args.logdir)
+    logger = get_logger(os.path.join(args.logdir, 'train_source.log'))
+    logger.info(args)
 
     # data
     source_transform = transforms.Compose([