diff --git a/train_model.py b/train_model.py
index 7cfa91c..0e6ce95 100644
--- a/train_model.py
+++ b/train_model.py
@@ -163,6 +163,8 @@ def main():
     total_frames_validation = average_frames_per_video * len(validation_videos)
     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.")