diff --git a/embedding_converter/README.md b/embedding_converter/README.md index 36aa0a5..399f9db 100644 --- a/embedding_converter/README.md +++ b/embedding_converter/README.md @@ -32,7 +32,7 @@ file_pattern = .datasets/megaface/**/*.jpg ``` [training.loader] -split_ratio = 0.8 +split_ratio = 0.95 batch_size = 256 num_workers = 8 ``` diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index 4b1f6b1..3d66360 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -111,8 +111,7 @@ def create_trainer() -> Trainer: save_top_k = 3, save_last = True ) - ], - val_check_interval = 100 + ] )