Skip to content
Snippets Groups Projects
Commit f13a8431 authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Fixed wrong positional argument in load method and keyword argument in plot_sample function

parent 12e39481
No related branches found
No related tags found
No related merge requests found
......@@ -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,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment