next set of run ymls

sig-gis · Apr 23, 2024 · 7e9db4f · 7e9db4f
1 parent c945536
commit 7e9db4f
Show file tree

Hide file tree

Showing 6 changed files with 76 additions and 4 deletions.
diff --git a/runc-mobilenetv3small-epochs10-batch32-lr001.yml b/runc-mobilenetv3small-epochs10-batch32-lr001.yml
@@ -0,0 +1,18 @@
+experiment_name: mobilenetv3small-epochs10-batch32-lr001
+model_name: mobilenet_v3small
+data_dir: tfrecords/train_test
+val_data_dir: tfrecords/val
+checkpoint: saved_models/mobilenetv3small-epochs10-batch32-lr001/best_model.h5
+total_examples: 70000 # number of geotiffs not tfrecords
+test_split: 0.2 # float or null
+
+optimizer: adam
+optimizer_use_lr_schedular: true
+loss_function: binary_crossentropy
+
+epochs: 10
+learning_rate: 0.001
+batch_size: 32
+buffer_size: 70000
+
+early_stopping_patience: 5 # null or int
diff --git a/runc-mobilenetv3small-epochs10-batch32-lr01.yml b/runc-mobilenetv3small-epochs10-batch32-lr01.yml
@@ -0,0 +1,18 @@
+experiment_name: mobilenetv3small-epochs10-batch32-lr01
+model_name: mobilenet_v3small
+data_dir: tfrecords/train_test
+val_data_dir: tfrecords/val
+checkpoint: saved_models/mobilenetv3small-epochs10-batch32-lr01/best_model.h5
+total_examples: 70000 # number of geotiffs not tfrecords
+test_split: 0.2 # float or null
+
+optimizer: adam
+optimizer_use_lr_schedular: true
+loss_function: binary_crossentropy
+
+epochs: 10
+learning_rate: 0.01
+batch_size: 32
+buffer_size: 70000
+
+early_stopping_patience: 5 # null or int
diff --git a/runc-mobilenetv3small-epochs10-batch64-lr001.yml b/runc-mobilenetv3small-epochs10-batch64-lr001.yml
@@ -0,0 +1,18 @@
+experiment_name: mobilenetv3small-epochs10-batch64-lr001
+model_name: mobilenet_v3small
+data_dir: tfrecords/train_test
+val_data_dir: tfrecords/val
+checkpoint: saved_models/mobilenetv3small-epochs10-batch64-lr001/best_model.h5
+total_examples: 70000 # number of geotiffs not tfrecords
+test_split: 0.2 # float or null
+
+optimizer: adam
+optimizer_use_lr_schedular: true
+loss_function: binary_crossentropy
+
+epochs: 10
+learning_rate: 0.001
+batch_size: 64
+buffer_size: 70000
+
+early_stopping_patience: 5 # null or int
diff --git a/runc-mobilenetv3small-epochs10-batch64-lr01.yml b/runc-mobilenetv3small-epochs10-batch64-lr01.yml
@@ -0,0 +1,18 @@
+experiment_name: mobilenetv3small-epochs10-batch64-lr01
+model_name: mobilenet_v3small
+data_dir: tfrecords/train_test
+val_data_dir: tfrecords/val
+checkpoint: saved_models/mobilenetv3small-epochs10-batch64-lr01/best_model.h5
+total_examples: 70000 # number of geotiffs not tfrecords
+test_split: 0.2 # float or null
+
+optimizer: adam
+optimizer_use_lr_schedular: true
+loss_function: binary_crossentropy
+
+epochs: 10
+learning_rate: 0.01
+batch_size: 64
+buffer_size: 70000
+
+early_stopping_patience: 5 # null or int
diff --git a/runc-resnet-epochs10-batch32-lr001.yml b/runc-resnet-epochs10-batch32-lr001.yml
@@ -1,8 +1,8 @@
-experiment_name: resnet-epochs10-batch64-lr001
+experiment_name: resnet-epochs10-batch32-lr001
 model_name: resnet
 data_dir: tfrecords/train_test
 val_data_dir: tfrecords/val
-checkpoint: saved_models/resnet-epochs10-batch64-lr001/best_model.h5
+checkpoint: saved_models/resnet-epochs10-batch32-lr001/best_model.h5
 total_examples: 70000 # number of geotiffs not tfrecords
 test_split: 0.2 # float or null
 

diff --git a/runc-resnet-epochs10-batch32-lr01.yml b/runc-resnet-epochs10-batch32-lr01.yml
@@ -1,8 +1,8 @@
-experiment_name: resnet-epochs10-batch64-lr01
+experiment_name: resnet-epochs10-batch32-lr01
 model_name: resnet
 data_dir: tfrecords/train_test
 val_data_dir: tfrecords/val
-checkpoint: saved_models/resnet-epochs10-batch64-lr01/best_model.h5
+checkpoint: saved_models/resnet-epochs10-batch32-lr01/best_model.h5
 total_examples: 70000 # number of geotiffs not tfrecords
 test_split: 0.2 # float or null