| from pathlib import Path | |
| import sys,numpy as np,torch | |
| R=Path(__file__).resolve().parents[1];sys.path.insert(0,str(R));from model.pysteps import * | |
| c=cfg(R);z=torch.load(R/c['paths']['checkpoint'],weights_only=True);m=STEPS(**z['model_config']);m.load_state_dict(z['model']);d=np.load(R/c['data']['path']);p=R/c['paths']['predictions'];p.parent.mkdir(parents=True,exist_ok=True);np.savez_compressed(p,ensemble=m(torch.tensor(d['input'])).detach(),target=d['target'],lead_minutes=np.arange(1,13)*5);print(p) | |