tztsai
commited on
Commit
·
d09d703
1
Parent(s):
c9a76e6
rename Fortran NN file
Browse files
qobsTTFFFFFTF30FFTFTF30TTFTFTFFF80FFTFTTF2699FFFF_X01_no_qp_no_adv_surf_F_Tin_qin_disteq_O_Trad_rest_Tadv_qadv_qout_qsed_RESCALED_7epochs_no_drop_REAL_NN_layers5in61out148_BN_F_te70.nc → NN_weights_YOG_convection.nc
RENAMED
File without changes
|
models.py
CHANGED
@@ -188,9 +188,7 @@ def load_from_netcdf_params(nc_file: str, dtype: str = "float32") -> ANN:
|
|
188 |
if __name__ == "__main__":
|
189 |
# Load the model from the netcdf file and save it to a checkpoint.
|
190 |
net = load_from_netcdf_params(
|
191 |
-
"
|
192 |
-
"surf_F_Tin_qin_disteq_O_Trad_rest_Tadv_qadv_qout_qsed_RESCALED_7epochs"
|
193 |
-
"_no_drop_REAL_NN_layers5in61out148_BN_F_te70.nc"
|
194 |
)
|
195 |
net.save("nn_state.pt")
|
196 |
print("Model saved to nn_state.pt")
|
|
|
188 |
if __name__ == "__main__":
|
189 |
# Load the model from the netcdf file and save it to a checkpoint.
|
190 |
net = load_from_netcdf_params(
|
191 |
+
"NN_weights_YOG_convection.nc"
|
|
|
|
|
192 |
)
|
193 |
net.save("nn_state.pt")
|
194 |
print("Model saved to nn_state.pt")
|