Merge pull request #84 from tsugumi-sys/delete-unnecessary-params

delete unnecessary parameters
tsugumi-sys · Jan 15, 2024 · 113ceec · 113ceec
2 parents 9883b9e + 7be8106
commit 113ceec
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
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

diff --git a/tests/pipelines/test_experimenter.py b/tests/pipelines/test_experimenter.py
@@ -16,7 +16,6 @@ def test_run():
         model = TestModel()
         training_params: TrainingParams = {
             "epochs": 1,
-            "batch_size": 1,
             "loss_criterion": nn.MSELoss(),
             "accuracy_criterion": nn.L1Loss(),
             "optimizer": Adam(model.parameters(), lr=0.0005),