credit.applications.rollout_to_netcdf_gen2

credit.applications.rollout_to_netcdf_gen2#

Rollout entry point for the v2 data schema (conf[“data”][“source”]).

Key differences from rollout_to_netcdf.py (v1):

  • No credit_main_parser / predict_data_check (those are v1-only)

  • Uses LocalDataset + build_preblocks/apply_preblocks for data loading

  • Inverse-normalization built from mean/std NC files (per-channel)

  • Injects flat data schema keys for output.py compatibility

  • forecast_len semantics: N steps at lead_time_periods hours per step

Usage:
torchrun –standalone –nnodes=1 –nproc-per-node=<N_GPUS>

applications/rollout_to_netcdf_gen2.py -c config/wxformer_025deg_6hr_gen2.yml

Or submit via PBS (see scripts/casper_gen2.sh with SCRIPT=applications/rollout_to_netcdf_gen2.py).

Attributes#

logger

Functions#

predict(rank, world_size, conf, p)

main()

Module Contents#

credit.applications.rollout_to_netcdf_gen2.logger#
credit.applications.rollout_to_netcdf_gen2.predict(rank, world_size, conf, p)#
credit.applications.rollout_to_netcdf_gen2.main()#