From 2b2424c9d9d3c16e1e5cfb3298a4ec412e47bc8d Mon Sep 17 00:00:00 2001 From: Eve-ning Date: Wed, 22 Nov 2023 14:24:13 +0800 Subject: [PATCH] Try debugging venv mounting --- tests/model_tests/chestnut_dec_may/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/model_tests/chestnut_dec_may/main.py b/tests/model_tests/chestnut_dec_may/main.py index 2d557015..3e2fda79 100644 --- a/tests/model_tests/chestnut_dec_may/main.py +++ b/tests/model_tests/chestnut_dec_may/main.py @@ -58,7 +58,7 @@ def train_val_test_split( labels = [*labels_0, *labels_1] BATCH_SIZE = 5 -EPOCHS = 15 +EPOCHS = 50 LR = 1e-3 # Prepare the datamodule and trainer