From 00376d009ca3caefe9e17c926350549a373f1e64 Mon Sep 17 00:00:00 2001 From: Yunchu Lee Date: Mon, 5 Aug 2024 14:07:00 +0900 Subject: [PATCH 1/2] bump torch version to 2.2.2 --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index fcb4f9c5ac3..e913a9df994 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -91,7 +91,7 @@ xpu = [ ] base = [ - "torch==2.1.1", + "torch==2.2.2", "lightning==2.2", "pytorchcv", "timm==1.0.3", From 9ee902913dd98b96ce5b9772f08d98dfbe6e2827 Mon Sep 17 00:00:00 2001 From: "Shin, Eunwoo" Date: Mon, 5 Aug 2024 15:31:03 +0900 Subject: [PATCH 2/2] update unit test --- tests/unit/cli/test_cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/cli/test_cli.py b/tests/unit/cli/test_cli.py index 9211e574db5..c464aababd2 100644 --- a/tests/unit/cli/test_cli.py +++ b/tests/unit/cli/test_cli.py @@ -164,7 +164,7 @@ def test_print_config_scheduler_override_command(self, fxt_print_config_schedule cooldown: 0 min_lr: 0.0 eps: 1.0e-08 - verbose: false + verbose: deprecated """ expected_config = yaml.safe_load(expected_str) assert expected_config["scheduler"] == result_config["model"]["init_args"]["scheduler"]