sup3r.models.multi_step

sup3r.models.multi_step#

Sup3r multi step model frameworks

TODO: SolarMultiStepGan can be cleaned up a little with the output padding and t_enhance argument moved to SolarCC.

Classes

MultiStepGan(models)

Multi-Step GAN, which is really just an abstraction layer on top of one or more Sup3rGan models that will perform their forward passes in serial.

MultiStepSurfaceMetGan(models)

A two-step GAN where the first step is a spatial-only enhancement on a 4D tensor of near-surface temperature and relative humidity data, and the second step is a (spatio)temporal enhancement on a 5D tensor.

SolarMultiStepGan(spatial_solar_models, ...)

Special multi step model for solar clearsky ratio super resolution.