optimizer = torch.optim.Adam(vgg16.parameters(), lr=LR) # optimize all cnn parameters criterion = nn.CrossEntropyLoss() # the target label is not one-hotted ...
確定！ 回上一頁
