flashΒΆ Input DataModule A basic DataModule class for all Flash tasks. FlashCallback FlashCallback is an extension of pytorch_lightning.callbacks.Callback. OutputTransform The OutputTransform encapsulates all the data processing logic that should run after the model. Output An Output encapsulates a single transform() method which is used to convert the model output into the desired output format when predicting. InputTransform Task A general Task. Trainer Exteded Trainer for FLash tasks.