From 0d8ff330294899063cd162541e06cf4b17399d61 Mon Sep 17 00:00:00 2001 From: Akira Date: Mon, 15 Jan 2024 23:09:09 +0900 Subject: [PATCH 1/2] delete unnecessary parameters --- pipelines/trainer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pipelines/trainer.py b/pipelines/trainer.py index 8860196..fda020a 100644 --- 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 From 7be810671be3563f184ab4f00a5a4e69c7948eee Mon Sep 17 00:00:00 2001 From: Akira Date: Tue, 16 Jan 2024 00:03:22 +0900 Subject: [PATCH 2/2] fix typping --- tests/pipelines/test_experimenter.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/pipelines/test_experimenter.py b/tests/pipelines/test_experimenter.py index 6d297a2..c03b4b5 100644 --- 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),