Shortcuts

FuncModule

class flash.core.data.utils.FuncModule(func)[source]

This class is used to wrap a callable within a nn.Module and apply the wrapped function in __call__