From 5f0fbbb1f866829fff4ae778b799dfa39078811e Mon Sep 17 00:00:00 2001 From: "Sergey E. Koposov" Date: Sat, 16 Nov 2024 21:29:48 +0000 Subject: [PATCH] update filenames --- py/rvspecfit/nn/train_interpolator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/rvspecfit/nn/train_interpolator.py b/py/rvspecfit/nn/train_interpolator.py index 6707b9c..f8fabe7 100644 --- 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,