ImageNetInputNorm

class torchelie.nn.ImageNetInputNorm

Normalize images channels as torchvision models expects, in a differentiable way

forward(input)
inverse(input)
training: bool