better formating

ain-soph · Dec 14, 2021 · 9254590 · 9254590
1 parent 8e513e2
commit 9254590
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/trojanzoo/models.py b/trojanzoo/models.py
@@ -521,8 +521,7 @@ def _train(self, epochs: int, optimizer: Optimizer,
                      save=save, amp=amp,
                      loader_train=loader_train, loader_valid=loader_valid,
                      epoch_fn=epoch_fn, get_data_fn=get_data_fn,
-                     loss_fn=loss_fn,
-                     after_loss_fn=after_loss_fn,
+                     loss_fn=loss_fn, after_loss_fn=after_loss_fn,
                      validate_fn=validate_fn,
                      save_fn=save_fn, file_path=file_path,
                      folder_path=folder_path, suffix=suffix,