Shortcuts

LogitsOutput

class flash.core.classification.LogitsOutput(multi_label=False)[source]

A Output which simply converts the model outputs (assumed to be logits) to a list.