ResNet

Create model:

model = ResNetModel(18; in_channels=3, classes=10)

or you can ommit classification layer, by specifying number of classes as nothing.

• simple:

y = x |> model

• extract list of features:

features = model(x, Val(:stages))

Number of channels for each element of features is in model.stages. Size of the model is in model.size.