小贤同学的博客

定义：#gpu or notdevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")使用：import torchimport torch.nn as nnimport torch.optim as optimimport torchvisionimport torchvision.tran...

以VGG模型为例：class LambdaBase(nn.Sequential): #lambda表达式 def __init__(self, fn, *args): super(LambdaBase, self).__init__(*args) self.lambda_func = fn def forward_prepare(sel...