rollout_samudra#
Attributes#
Classes#
Functions#
Module Contents#
- rollout_samudra.logger#
- class 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)#
- rollout_samudra.predict(rank, world_size, conf, p)#
- rollout_samudra.main()#