Debug

class torchelie.nn.Debug(name)

An pass-through layer that prints some debug info during forward pass. It prints its name, the input’s shape, mean of channels means, mean, mean of channels std, and std.

Parameters

name (str) – this layer’s name

forward(x)

training: bool