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

Changed batch size

Browse source
This commit is contained in:
Jordon Brooks 2023-07-26 00:12:36 +01:00
parent 0c2d4a5ce7
commit 8c5001166d
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
train_model.py
View file

@ -9,9 +9,9 @@ from tensorflow.keras.callbacks import EarlyStopping
print(tf.config.list_physical_devices('GPU')) print(tf.config.list_physical_devices('GPU'))
# Constants # Constants
BATCH_SIZE = 16 BATCH_SIZE = 8
EPOCHS = 1 EPOCHS = 5
TRAIN_SAMPLES = 1 TRAIN_SAMPLES = 5
def load_list(list_path): def load_list(list_path):
with open(list_path, "r") as json_file: with open(list_path, "r") as json_file: