Added gc.collect() in the main function

2023-08-17 23:40:56 +01:00 · 2023-08-17 23:40:56 +01:00 · d3f1ac1e0d
commit d3f1ac1e0d
parent 7787d0584e
1 changed files with 2 additions and 0 deletions
--- a/train_model.py
+++ b/train_model.py
@ -164,6 +164,8 @@ def main():
    steps_per_epoch_train = total_frames_train // BATCH_SIZE
    steps_per_epoch_validation = total_frames_validation // BATCH_SIZE
    
+    gc.collect()
+
    # Train the model
    LOGGER.info("Starting model training.")
    model.fit(