From 9fbb3e8b25d625a70636f42188ba7a6854b43d83 Mon Sep 17 00:00:00 2001
From: Francesco Vaselli <francesco.vaselli@protonmail.com>
Date: Wed, 22 May 2024 11:48:40 +0200
Subject: [PATCH] different splits

---
 configs/new_train_config.yaml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/configs/new_train_config.yaml b/configs/new_train_config.yaml
index 6b85334..da9baf5 100644
--- 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