Commit b806a53a authored by sjjsmuel's avatar sjjsmuel

train_params

parent f23cafb1
......@@ -53,10 +53,11 @@ img_width = options.width
img_height = options.height
channels = 3
n_classes = 2
batch_size = 16
batch_size = options.batch_size
batch_count_train_dataset = 2000 // batch_size
batch_count_validation_dataset = 800 // batch_size
batch_count_test_dataset = 800 // batch_size
batch_count_validation_dataset = 500 // batch_size
batch_count_test_dataset = 500 // batch_size
# Load the dataset into TF Datasets
# Training Data
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment