Skip to content
Snippets Groups Projects
Commit 5427d7e9 authored by Bobholamovic's avatar Bobholamovic
Browse files

Fix extra channel in a saved image

parent 134eddfd
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
config_base.yaml
+ 1
− 1
View file @ 5427d7e9
...@@ -19,7 +19,7 @@ step: 2 ...@@ -19,7 +19,7 @@ step: 2
# Training related # Training related
batch_size: 32 batch_size: 32
num_epochs: 20 num_epochs: 40
resume: '' resume: ''
load_optim: True load_optim: True
anew: False anew: False
......
src/core/trainers.py
+ 1
− 1
View file @ 5427d7e9
...@@ -267,6 +267,6 @@ class CDTrainer(Trainer): ...@@ -267,6 +267,6 @@ class CDTrainer(Trainer):
self.logger.dump(desc) self.logger.dump(desc)
if store: if store:
self.save_image(name[0], CM, epoch) self.save_image(name[0], CM.squeeze(-1), epoch)
return self.metrics[0].avg if len(self.metrics) > 0 else max(1.0 - losses.avg, self._init_max_acc) return self.metrics[0].avg if len(self.metrics) > 0 else max(1.0 - losses.avg, self._init_max_acc)
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment