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

Added gc.collect() in the main function

Browse source
This commit is contained in:
Jordon Brooks 2023-08-17 23:40:56 +01:00
parent 7787d0584e
commit d3f1ac1e0d
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

@ -163,6 +163,8 @@ def main():
total_frames_validation = average_frames_per_video * len(validation_videos) total_frames_validation = average_frames_per_video * len(validation_videos)
steps_per_epoch_train = total_frames_train // BATCH_SIZE steps_per_epoch_train = total_frames_train // BATCH_SIZE
steps_per_epoch_validation = total_frames_validation // BATCH_SIZE steps_per_epoch_validation = total_frames_validation // BATCH_SIZE
gc.collect()
# Train the model # Train the model
LOGGER.info("Starting model training.") LOGGER.info("Starting model training.")