记录分类模型,对应参数准确率和损失下降情况
densenet 加入drop out p=0.4
sgd优化器
optimizer = torch.optim.SGD(classify_model.parameters(), lr=params["lr"], momentum=0.9, weight_decay=1e-5)
scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer, mode='min', factor=0.7, patience=2)
dense...
原创
2021-04-02 10:17:22 ·
1237 阅读 ·
0 评论