Commit e3bc6ed6 authored by sjjsmuel's avatar sjjsmuel

param update

parent b806a53a
......@@ -57,6 +57,9 @@ batch_size = options.batch_size
batch_count_train_dataset = 2000 // batch_size
batch_count_validation_dataset = 500 // batch_size
batch_count_test_dataset = 500 // batch_size
batch_count_train_dataset = 200 // batch_size
batch_count_validation_dataset = 50 // batch_size
batch_count_test_dataset = 50 // batch_size
# Load the dataset into TF Datasets
......
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