OutputTransform¶
- class flash.core.data.io.output_transform.OutputTransform(running_stage=None)[source]¶
The
OutputTransform
encapsulates all the data processing logic that should run after the model.- static per_batch_transform(batch)[source]¶
Transforms to apply on a whole batch before uncollation to individual samples.
Can involve both CPU and Device transforms as this is not applied in separate workers.
- Return type