change pretrain to pretrained?

[CarloLucibello]

Should we have

VGG(11, pretrained=true)

instead of

VGG(11, pretrain=true)

The latter seems a command...

[CarloLucibello]

Even better, as in torchvision we could have a weights argument instead where we can pass more specific values, in case of multiple pretraining options:

VGG(11, weights="IMAGENET1K_V1")

[darsnack]

Yes the loading logic already supports that, and the plan was to move to that once we have multiple weight options.