fixed tests (#1787)

Co-authored-by: Eugene Khvedchenya <ekhvedchenya@gmail.com>
Deci-AI · Jan 25, 2024 · ca9a89b · ca9a89b
1 parent 8cb938a
commit ca9a89b
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/tests/recipe_training_tests/coded_qat_launch_test.py b/tests/recipe_training_tests/coded_qat_launch_test.py
@@ -15,9 +15,7 @@ def test_qat_launch(self):
         net = ResNet18(num_classes=10, arch_params={})
         train_params = {
             "max_epochs": 10,
-            "lr_updates": [],
             "lr_decay_factor": 0.1,
-            "lr_mode": "StepLRScheduler",
             "lr_warmup_epochs": 0,
             "initial_lr": 0.1,
             "loss": "CrossEntropyLoss",
@@ -76,9 +74,7 @@ def test_ptq_launch(self):
         net = ResNet18(num_classes=10, arch_params={})
         train_params = {
             "max_epochs": 10,
-            "lr_updates": [],
             "lr_decay_factor": 0.1,
-            "lr_mode": "StepLRScheduler",
             "lr_warmup_epochs": 0,
             "initial_lr": 0.1,
             "loss": "CrossEntropyLoss",

diff --git a/tests/unit_tests/test_lr_assignment.py b/tests/unit_tests/test_lr_assignment.py
@@ -138,9 +138,7 @@ def test_train_with_lr_assignment(self):
 
         train_params = {
             "max_epochs": 3,
-            "lr_updates": [],
             "lr_decay_factor": 0.1,
-            "lr_mode": "StepLRScheduler",
             "initial_lr": {
                 "default": 0,
                 "fc3": 0.1,