模型训练坎坷路--逐步提升模型准确率从40%到90%+
剪枝实验的基准得是一个微调好的准确率保持较高的模型,按照我之前yolov5的经验来看,这还不简单?选模型+合适的数据集,20轮就OK!我计划使用 ResNet-18模型加上Cifar-10数据集,训练轮次100轮,准确度达到95%我就立马停下。可是过程却不大对劲,都20轮了我的准确率还是40%上下窜,一点进步的痕迹都没有,于是,果断按下停止键…from torchvision import models # 引入torchvision.modelstry:def train(
原创
2024-08-15 14:48:17 ·
526 阅读 ·
0 评论