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

Added support for continued model training

Browse source
This commit is contained in:
Jordon Brooks 2023-08-13 14:21:47 +01:00
parent fde856f3ec
commit f43ef69f0d
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
train_model.py
View file

@ -90,7 +90,11 @@ def main():
training_videos = all_videos[:split_index] training_videos = all_videos[:split_index]
validation_videos = all_videos[split_index:] validation_videos = all_videos[split_index:]
model = VideoCompressionModel() if args.continue_training:
model = tf.keras.models.load_model(args.continue_training)
else:
model = VideoCompressionModel()
# Set optimizer and compile the model # Set optimizer and compile the model
optimizer = tf.keras.optimizers.Adam(learning_rate=LEARNING_RATE) optimizer = tf.keras.optimizers.Adam(learning_rate=LEARNING_RATE)