search

MegEngine / Models

采用MegEngine实现的各种主流深度学习模型
Other
303 stars 99 forks source link

resnet分类模型无法指定分类数量 #115

Open Qianlu5110 opened 3 years ago

Qianlu5110 commented 3 years ago

使用 classification resnet进行分类苹果/橘子训练时，指定分类数量为2时，无法训练

Models/official/vision/classification/resnet/model.py resnet50只支持1000分类数据集训练，我只创建2分类数据时，修改num_classes=2时，无法正常完成训练

任务描述

数据集格式是这样的： /path/to/imagenet train apple xxx.jpg ... orangle xxx.jpg ... ... val apple xxx.jpg ... orangle xxx.jpg ...

目标

由于我时初学，在经过自己各种修改后，仍然无法解决，需要帮忙。谢谢