From 58c38ae5f1279b1a714c75d0c8da089412b60231 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Tue, 4 Oct 2022 09:03:10 +0200
Subject: [PATCH] default training now logs args
---
mu_map/training/default.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/mu_map/training/default.py b/mu_map/training/default.py
index 90ab158..6ef94d9 100644
--- a/mu_map/training/default.py
+++ b/mu_map/training/default.py
@@ -205,6 +205,8 @@ if __name__ == "__main__":
device = torch.device(args.device)
logger = get_logger_by_args(args)
+ logger.info(args)
+
model = UNet(in_channels=1, features=args.features)
model = model.to(device)
--
GitLab