[USGS-R#146] rm print statement; move comment

jsadler2 · Jan 19, 2022 · 64a9ea0 · 64a9ea0
1 parent 454dc73
commit 64a9ea0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/river_dl/train.py b/river_dl/train.py
@@ -82,7 +82,7 @@ def train_model(
         csv_log = tf.keras.callbacks.CSVLogger(log_file)
         callbacks.append(csv_log)
 
-    print(best_val_weight_dir)
+    # Save alternate weight file that saves the best validation weights
     if best_val_weight_dir:
         best_val = tf.keras.callbacks.ModelCheckpoint(
             best_val_weight_dir, monitor='val_loss', verbose=0, save_best_only=True,
@@ -130,6 +130,5 @@ def train_model(
     if weight_dir:
         model.save_weights(weight_dir)
 
-    # Save alternate weight file that saves the best validation weights
 
     return model