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

Changed video and model paths to new location

Browse source
This commit is contained in:
Jordon Brooks 2023-07-24 16:51:56 +01:00
parent 58fcf819ee
commit 80c5f2216d
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
train_model.py
View file

@ -10,8 +10,8 @@ BATCH_SIZE = 32 # Batch size used during training
EPOCHS = 20 # Number of training epochs
# Step 1: Data Preparation
TRAIN_VIDEO_FILE = 'native_video.mkv' # The training video file name
VAL_VIDEO_FILE = 'training_video.mkv' # The validation video file name
TRAIN_VIDEO_FILE = 'test_data/native_video.mkv' # The training video file name
VAL_VIDEO_FILE = 'test_data/training_video.mkv' # The validation video file name
TRAIN_SAMPLES = 2 # Number of video frames used for training
VAL_SAMPLES = 2 # Number of video frames used for validation
@ -77,5 +77,5 @@ model.fit(
print("\nTraining completed.")
# Step 3: Save the trained model
model.save('ai_rate_control_model.keras')
model.save('models/model.keras')
print("Model saved successfully!")