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")
|