From 827b5761f677f796c9e5e46735f07d6ef173fc48 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Wed, 3 Mar 2021 16:52:35 +0100
Subject: [PATCH] Initialize logging with model name.
---
pysegcnn/main/train_transfer.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/pysegcnn/main/train_transfer.py b/pysegcnn/main/train_transfer.py
index 610d7ac..4220ecf 100644
--- a/pysegcnn/main/train_transfer.py
+++ b/pysegcnn/main/train_transfer.py
@@ -94,6 +94,7 @@ if __name__ == '__main__':
# (xiii) instanciate logging configuration
net_lc = LogConfig(state_file)
dictConfig(log_conf(net_lc.log_file))
+ net_lc.init_log('Initializing model: {}'.format(state_file.name))
# (xiv) instanciate the model
if trn_sf.supervised or trn_sf.uda_from_pretrained:
@@ -117,7 +118,7 @@ if __name__ == '__main__':
trainer = DomainAdaptationTrainer(
model=net,
optimizer=optimizer,
- state_file=state_file,
+ state_file=net.state_file,
src_train_dl=src_tra_dl,
src_valid_dl=src_val_dl,
src_test_dl=src_tes_dl,
--
GitLab