Added support for continued model training
This commit is contained in:
parent
fde856f3ec
commit
f43ef69f0d
1 changed files with 5 additions and 1 deletions
|
@ -90,7 +90,11 @@ def main():
|
|||
training_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
|
||||
optimizer = tf.keras.optimizers.Adam(learning_rate=LEARNING_RATE)
|
||||
|
|
Reference in a new issue