From bfa8caa032e05fec85985a53c4c62c6467e1d73d Mon Sep 17 00:00:00 2001
From: Jordon Brooks <jordonbc@hotmail.co.uk>
Date: Wed, 26 Jul 2023 00:11:57 +0100
Subject: [PATCH] Added GPU debug test

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

diff --git a/train_model.py b/train_model.py
index ce0d2a5..f08b84e 100644
--- a/train_model.py
+++ b/train_model.py
@@ -6,6 +6,8 @@ import cv2
 from video_compression_model import NUM_CHANNELS, NUM_FRAMES, VideoCompressionModel, PRESET_SPEED_CATEGORIES
 from tensorflow.keras.callbacks import EarlyStopping
 
+print(tf.config.list_physical_devices('GPU'))
+
 # Constants
 BATCH_SIZE = 16
 EPOCHS = 1