sup3r.pipeline.strategy

ForwardPassStrategy class. This sets up chunks and needed generator inputs to distribute forward passes.

Classes

ForwardPassChunk(input_data, exo_data, ...)	Structure storing chunk data and attributes for a specific chunk going through the generator.
ForwardPassStrategy(file_paths, model_kwargs)	Class to prepare data for forward passes through generator.