different splits

francesco-vaselli · May 22, 2024 · 9fbb3e8 · 9fbb3e8
1 parent 19958ed
commit 9fbb3e8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/configs/new_train_config.yaml b/configs/new_train_config.yaml
@@ -4,9 +4,9 @@ data_mean: 144.96 # 144.982 # 127.836
 data_std:  58.062 # 57.9408 # 60.410
 # 57.940860748291016
 transfer_data_path: "./data/dataset_99908129.npy"
-n_train: 500000
-n_val: 300000
-n_test: 500000
+n_train: 1000000
+n_val: 1000000
+n_test: 1000000
 batch_size: 64
 buffer_size: 1000 # not used
 epochs: 200