credit.applications.rollout_samudra#

Attributes#

Classes#

Functions#

predict(rank, world_size, conf, p)

main()

Module Contents#

credit.applications.rollout_samudra.logger#
class credit.applications.rollout_samudra.ForecastProcessor(conf, state_transformer, device)#
conf#
device#
batch_size#
ensemble_size#
lead_time_periods#
state_transformer#
latlons#
meta_data#
process(y_pred, forecast_step, forecast_count, datetimes, save_datetimes)#
credit.applications.rollout_samudra.predict(rank, world_size, conf, p)#
credit.applications.rollout_samudra.main()#