fix two configs and remove a print in model_fit

sig-gis · Apr 23, 2024 · b957d70 · b957d70
1 parent 3b93e87
commit b957d70
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/fao_models/model_fit.py b/fao_models/model_fit.py
@@ -156,7 +156,6 @@ def main():
     logger.info("Training history:")
     logger.info(pformat(history.history))
 
-    print(val_data_dir)
     if val_data_dir:
         val_dataset = dl.load_dataset_from_tfrecords(val_data_dir, batch_size=batch_size)
         eval = model.evaluate(val_dataset,return_dict=True)

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-lr01
+experiment_name: resnet-epochs10-batch64-lr001
 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-batch64-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-batch64-lr001.yml b/runc-resnet-epochs10-batch64-lr001.yml
@@ -1,8 +1,8 @@
-experiment_name: resnet-epochs10-batch64-lr01
+experiment_name: resnet-epochs10-batch64-lr001
 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-batch64-lr001/best_model.h5
 total_examples: 70000 # number of geotiffs not tfrecords
 test_split: 0.2 # float or null