rollout_to_netcdf

rollout_to_netcdf#

Attributes#

logger

description

Classes#

ForecastProcessor

Functions#

predict(rank, world_size, conf, p)

Module Contents#

rollout_to_netcdf.logger#
class rollout_to_netcdf.ForecastProcessor(conf, device)#
conf#
device#
batch_size#
ensemble_size#
lead_time_periods#
latlons#
meta_data#
process(y_pred, forecast_step, forecast_count, datetimes, save_datetimes)#
rollout_to_netcdf.predict(rank, world_size, conf, p)#
rollout_to_netcdf.description = 'Rollout AI-NWP forecasts'#
Contents