From 7404385c314059e7f8574b1c0cb0cdec5b3f6b62 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Wed, 3 Mar 2021 16:49:24 +0100
Subject: [PATCH] Use correct state file to save network configuration.
---
pysegcnn/main/train_transfer.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/pysegcnn/main/train_transfer.py b/pysegcnn/main/train_transfer.py
index dfb9f36..610d7ac 100644
--- a/pysegcnn/main/train_transfer.py
+++ b/pysegcnn/main/train_transfer.py
@@ -110,11 +110,14 @@ if __name__ == '__main__':
net, optimizer, checkpoint = net_mc.init_model(
len(src_ds.use_bands), len(src_ds.labels), state_file)
+ # set the model state file
+ net.state_file = state_file
+
# (xv) instanciate the network trainer class
trainer = DomainAdaptationTrainer(
model=net,
optimizer=optimizer,
- state_file=net.state_file,
+ state_file=state_file,
src_train_dl=src_tra_dl,
src_valid_dl=src_val_dl,
src_test_dl=src_tes_dl,
--
GitLab