update filenames

segasai · Nov 16, 2024 · 5f0fbbb · 5f0fbbb
1 parent 518e207
commit 5f0fbbb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py/rvspecfit/nn/train_interpolator.py b/py/rvspecfit/nn/train_interpolator.py
@@ -288,7 +288,7 @@ def main(args):
 
     torch.save(myint.state_dict(), finalfile_path)
     pred = get_predictions(myint, Tvecs0, train_dev, batch)
-    cur_name = f'{directory}/pred_{setup}.psav'
+    cur_name = f'{directory}/pred_{setup}.h5'
     DD = {}
     DD['pred'] = pred,
     DD['vecs'] = vecs
@@ -301,7 +301,7 @@ def main(args):
     import rvspecfit.nn.RVSInterpolator  # noqa
     mapper_module = 'rvspecfit.nn.NNInterpolator'
     mapper_class_name = 'Mapper'
-    ofname = f'{directory}/interp_{setup}.pkl'
+    ofname = f'{directory}/interp_{setup}.h5'
     D = {
         'mapper_module': mapper_module,
         'mapper_class_name': mapper_class_name,