diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index b77e3a5..ad69e97 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -106,7 +106,8 @@ def create_trainer() -> Trainer: save_top_k = 3, save_last = True ) - ] + ], + val_check_interval = 10 ) diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 89fa662..1ca38ba 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -118,7 +118,7 @@ def create_trainer() -> Trainer: save_last = True ) ], - val_check_interval = 1000, + val_check_interval = 1000 )