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__': ...@@ -43,7 +43,8 @@ if __name__ == '__main__':
if trainer.plot_samples: if trainer.plot_samples:
# load pretrained model # 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.state_path)
trainer.model.eval() trainer.model.eval()
...@@ -79,7 +80,7 @@ if __name__ == '__main__': ...@@ -79,7 +80,7 @@ if __name__ == '__main__':
labels, labels,
trainer.dataset.use_bands, trainer.dataset.use_bands,
trainer.dataset.labels, trainer.dataset.labels,
ypred=y_pred, y_pred=y_pred,
bands=trainer.plot_bands, bands=trainer.plot_bands,
state=sname, state=sname,
stretch=True, stretch=True,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment