jordon/DeepEncode
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added GPU debug test

Browse source
This commit is contained in:
Jordon Brooks 2023-07-26 00:11:57 +01:00
parent f390b585f9
commit bfa8caa032
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
train_model.py
View file

@ -6,6 +6,8 @@ import cv2
from video_compression_model import NUM_CHANNELS, NUM_FRAMES, VideoCompressionModel, PRESET_SPEED_CATEGORIES from video_compression_model import NUM_CHANNELS, NUM_FRAMES, VideoCompressionModel, PRESET_SPEED_CATEGORIES
from tensorflow.keras.callbacks import EarlyStopping from tensorflow.keras.callbacks import EarlyStopping
print(tf.config.list_physical_devices('GPU'))
# Constants # Constants
BATCH_SIZE = 16 BATCH_SIZE = 16
EPOCHS = 1 EPOCHS = 1