https://github.com/lapis-zero09/pytorchlight

pytorchlightの使いかた

https://github.com/lapis-zero09/pytorchlight

how to use vgg

import cv
model = cv.vgg11(in_channels=1, num_classes=10).cuda()
model

see also: https://github.com/lapis-zero09/pytorchlight/blob/master/lib/lib_test.ipynb