-
Notifications
You must be signed in to change notification settings - Fork 0
/
export_model.py
31 lines (19 loc) · 1.33 KB
/
export_model.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from gaia.export import export
from datetime import date
# model_dir = "lightning_logs_transformer/version_10" #best model on old data
model_dir = "lightning_logs/version_0"
timestamp = date.today().strftime("%m-%d-%y")
export_name = f"export_model_cam4_{timestamp}.pt"
# inputs = ['FLNT', 'FLNS', 'T', 'SOLIN', 'Z3', 'OMEGA', 'FSNS', 'LHFLX', 'FSNT', 'SHFLX', 'U', 'V', 'Q', 'PSL']
# outputs = ['PRECC', 'PTTEND', 'PTEQ', 'Z3', 'PRECT']
# inputs = "Q,T,U,V,OMEGA,Z3,PS,SOLIN,SHFLX,LHFLX,FSNS,FLNS,FSNT,FLNT,FSDS".split(",")
# outputs = "PTEQ,PTTEND,DCQ,DTCOND,QRS,QRL,CLOUD,CONCLD,FSNS,FLNS,FSNT,FLNT,FSDS,PRECT,PRECC,PRECL,PRECSC,PRECSL".split(",")
### v2
# Q,T,U,V,OMEGA,Z3,PS,SOLIN,SHFLX,LHFLX,FSNS,FLNS,FSNT,FLNT,FSDS
# CLOUD,CONCLD,DCQ,DTCOND,FLDS,PRECC,PRECL,PRECSC,PRECSL,PRECT,PSL,PTEQ,PTTEND,QRL,QRS,SOLL,SOLLD,SOLS,SOLSD,SRFRAD
# inputs = None
# outputs = "PTEQ,PTTEND,DCQ,DTCOND,QRS,QRL,CLOUD,CONCLD,FSNS,FLNS,FSNT,FLNT,FSDS,FLDS,SRFRAD,SOLL,SOLS,SOLLD,SOLSD,PSL,PRECT,PRECC,PRECL,PRECSC,PRECSL".split(",")
# desired_inputs =
inputs = "Q,T,U,V,OMEGA,Z3,PS,SOLIN,SHFLX,LHFLX,FSNS,FLNS,FSNT,FLNT,FSDS".split(",")
outputs = "PTEQ,PTTEND,DCQ,DTCOND,QRS,QRL,CLOUD,CONCLD,FSNS,FLNS,FSNT,FLNT,FSDS,FLDS,SRFRAD,SOLL,SOLS,SOLLD,SOLSD,PSL,PRECT,PRECC,PRECL,PRECSC,PRECSL".split(",")
export(model_dir, export_name, inputs=inputs, outputs=outputs)