File size: 562 Bytes
61c016d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | from pathlib import Path
import sys
ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))
from model.gcrm_convectionnn import load_yaml, run_inference
if __name__ == "__main__":
config = load_yaml(ROOT / "conf/config.yaml")
result = run_inference(str(ROOT / config["paths"]["data"]), str(ROOT / config["paths"]["checkpoint"]),
str(ROOT / config["paths"]["predictions"]), config)
print(f"saved inputs={result['inputs'].shape}, predictions={result['prediction'].shape}, lrf={result['lrf_full'].shape}")
|