chickadee » nanograd-components » make-am-resnet

make-am-resnet stages widths num-classes #!key (in-channels 3) (stem 'cifar) (dtype 'f32) (name AM-ResNet)procedure

Creates the lazy model. stem: 'strided gives the 3x3 stem stride 2 instead of 1.