From f13a8431207139b7a599a26c70a9d06154d934c1 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 16 Jul 2020 13:21:24 +0000
Subject: [PATCH] Fixed wrong positional argument in load method and keyword
argument in plot_sample function
---
main/eval.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/main/eval.py b/main/eval.py
index 3d6d521..da3fa14 100755
--- a/main/eval.py
+++ b/main/eval.py
@@ -43,7 +43,8 @@ if __name__ == '__main__':
if trainer.plot_samples:
# load pretrained model
- state = trainer.model.load(trainer.optimizer, trainer.state_file,
+ state = trainer.model.load(trainer.state_file,
+ trainer.optimizer,
trainer.state_path)
trainer.model.eval()
@@ -79,7 +80,7 @@ if __name__ == '__main__':
labels,
trainer.dataset.use_bands,
trainer.dataset.labels,
- ypred=y_pred,
+ y_pred=y_pred,
bands=trainer.plot_bands,
state=sname,
stretch=True,
--
GitLab