Skip to content

Commit

Permalink
netowrk to network
Browse files Browse the repository at this point in the history
  • Loading branch information
camillebrianceau committed Sep 3, 2024
1 parent 315d452 commit bb61475
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions clinicadl/trainer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -764,7 +764,7 @@ def _train(
amp=self.config.computational.amp,
)
criterion = get_criterion(
self.maps_manager.netowrk_task, self.config.model.loss
self.maps_manager.network_task, self.config.model.loss
)

optimizer = self._init_optimizer(model, split=split, resume=resume)
Expand All @@ -789,7 +789,7 @@ def _train(
if cluster.master:
log_writer = LogWriter(
self.maps_manager.maps_path,
evaluation_metrics(self.maps_manager.netowrk_task) + ["loss"],
evaluation_metrics(self.maps_manager.network_task) + ["loss"],
split,
resume=resume,
beginning_epoch=beginning_epoch,
Expand Down Expand Up @@ -1111,7 +1111,7 @@ def _train_ssdann(

log_writer = LogWriter(
self.maps_manager.maps_path,
evaluation_metrics(self.maps_manager.netowrk_task) + ["loss"],
evaluation_metrics(self.maps_manager.network_task) + ["loss"],
split,
resume=resume,
beginning_epoch=beginning_epoch,
Expand Down

0 comments on commit bb61475

Please sign in to comment.