Update my_model/utilities.py

my_model/utilities.py

@@ -8,6 +8,7 @@ import torch
 import matplotlib.pyplot as plt
 from IPython import get_ipython
 import sys
+import gc
 
 
 class VQADataProcessor:
@@ -290,6 +291,22 @@ def get_model_path(model_name):
     return model_path
 
 
+def free_gpu_resources():
+    """
+    Clears GPU memory.
+    """
+    try:
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+            gc.collect()
+            torch.cuda.empty_cache()
+            gc.collect()
+
+            print("GPU memory has been cleared.")
+    except Exception as e:
+        print(f"Error occurred while clearing GPU memory: {str(e)}")
+
+
 
 if __name__ == "__main__":
     pass