Fixing model checkpoint naming compatibility

dev-geof · Nov 7, 2024 · 7680687 · 7680687
1 parent e509c55
commit 7680687
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/validation.py b/src/validation.py
@@ -350,7 +350,7 @@ def transformer_validation(
     # Save predictions into hdf5
     if save_predictions == True:
 
-        f = h5py.File(str(model_path) + "/validation/prediction.h5", "w")
+        f = h5py.File(str(model_path) + "/validation/prediction.keras", "w")
 
         for d in range(len(val_sample_list)):