From d3f1ac1e0d4d5e2cb3cac351f990cadda794f763 Mon Sep 17 00:00:00 2001
From: Jordon Brooks <jordonbc@hotmail.co.uk>
Date: Thu, 17 Aug 2023 23:40:56 +0100
Subject: [PATCH] Added gc.collect() in the main function

---
 train_model.py | 2 ++
 1 file changed, 2 insertions(+)

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.")