如何修改pytorch官方实现的Densenet的分类数，用于跑自己的数据

pytorch官方实现的Densenet的分类数是1000类，当我们想用来跑自己的数据集时，类别可能没有这么多，这时就需要我们去更改网络最后的分类器

##修改分类器代码如下

model = models.densenet169(pretrained=True)
###重塑最后分类层  class_num为自己的类别数，可以直接填数字
model.classifier = torch.nn.Linear(model.classifier.in_features, class_num) 
model = model.train()