delete unnecessary parameters

tsugumi-sys · Jan 15, 2024 · 0d8ff33 · 0d8ff33
1 parent 9883b9e
commit 0d8ff33
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/pipelines/trainer.py b/pipelines/trainer.py
@@ -16,7 +16,6 @@
 
 class TrainingParams(TypedDict):
     epochs: int
-    batch_size: int
     loss_criterion: nn.Module
     accuracy_criterion: nn.Module
     optimizer: nn.Module