1.pytorch:models.需要的网络
for parameters in model.parameters():
parameters.require_grad=false #不参与训练
指定某些层不参与训练:待更
model.fc=model.classifier=Classifier pytorch中含义一样,可能版本不同
models是torchvision里面的
作业:finetuning,猫狗分类 ddl:1201
1.pytorch:models.需要的网络
for parameters in model.parameters():
parameters.require_grad=false #不参与训练
指定某些层不参与训练:待更
model.fc=model.classifier=Classifier pytorch中含义一样,可能版本不同
models是torchvision里面的
作业:finetuning,猫狗分类 ddl:1201