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

[Qianlu5110]

使用 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
...

目标

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