Krischal's Digital Garden

Initializing Lazy Layers in Pytorch

1 min read

A way to initialize model’s lazy layers.

model(torch.zeros([*input_shape]))

Graph View