- 博客(2)
- 收藏
- 关注
RSS订阅原创 size mismatch for head.weight: copying a param with shape torch.Size([1000, 768]) from checkpoint
我在github上的vit分类模型（）使用了自己的数据集，训练的时候发生报错。报错原因：训练模型时出现FC层不匹配，我的数据集是 3个类的，下载的预权重是ImageNet 1000个类的。
2024-08-01 10:16:18
378
yolov5疲劳驾驶数据集
疲劳驾驶数据集，标签格式为txt格式，分为训练集：验证集=8:2（一共2914张图片）
类别为：closed_eye、closed_mouth、open_eye、open_mouth
2023-05-17
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人