feature: minor renaming

Flippchen · Feb 15, 2024 · 22c3c67 · 22c3c67
1 parent e35860e
commit 22c3c67
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/training/train.py b/training/train.py
@@ -47,7 +47,7 @@
     "seed": random.randint(0, 1000) if random_seed else 123
 }
 tune_model = False
-tune_and_train = False
+train = True
 
 # Load dataset and classes
 train_ds, val_ds, class_names = load_dataset(**config)
@@ -136,7 +136,7 @@
     tuner.search(train_ds, epochs=5, validation_data=val_ds)
     model = tuner.get_best_models()[0]
 
-if tune_and_train or not tune_model:
+if train or not tune_model:
     with tf.device(device):
         history = model.fit(
             train_ds,