chore: change name of ann scripted model

DataWaveProject · Feb 5, 2025 · 67cad7c · 67cad7c
1 parent 4f8f619
commit 67cad7c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/era5_training/infer.f90 b/era5_training/infer.f90
@@ -69,7 +69,7 @@ program infer
     call torch_tensor_from_array(output_tensor(1), prediction_2d, layout_2d, torch_kCPU)
 
     ! Load ML model
-    call torch_model_load(model, trim(script_model_path) // "/nlgw_ann_gpu_scripted.pt", device_type=torch_kCUDA, device_index=0)
+    call torch_model_load(model, trim(script_model_path) // "/nlgw_ann-cnn_gpu_scripted.pt", device_type=torch_kCUDA, device_index=0)
 
     ! Infer
     do i = 1, num_loops

diff --git a/utils/function_training.py b/utils/function_training.py
@@ -205,7 +205,7 @@ def Inference_and_Save_ANN_CNN(model, testset, testloader, bs_test, device, sten
         # dummy_inputs = torch.tensor(np.ones(INP.shape, dtype=np.float32)).to(device)
         # trace_to_torchscript(model, dummy_input=dummy_inputs, filename="nlgw_ann_gpu_traced.pt")
         print("scripting...")
-        script_to_torchscript(model, filename="nlgw_ann_gpu_scripted.pt")
+        script_to_torchscript(model, filename="nlgw_ann-cnn_gpu_scripted.pt")
         print("complete")
 
         S = PRED.shape