A LightningModule organizes your PyTorch code into 6 sections: ... for model in [resnet50(), vgg16(), BidirectionalRNN()]: task = ClassificationTask(model) ...
確定! 回上一頁