记录分类模型,对应参数准确率和损失下降情况
densenet 加入drop out p=0.4sgd优化器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 ·
1211 阅读 ·
0 评论