关于多分类的问题

[kaixinbaba]

代码中和训练有关的代码反复看了好几遍，算大致搞懂了流程，想基于这个模型做点调整，代码中文本分类只有两类，文字和背景，我想修改成文字多分类，比如有10类，维度什么就跟着调整了下从原来的2 变成了11，但是训练开始没多久loss就开始无限增大了，估计是和修改的地方有关系 。自己折腾了很久也没看出哪里不对。。大佬有什么建议么

有个疑问 一开始的这个Input 参数2 代表的是2个分类还是第一索引是分类，第二索引是padding？

gt_class_ids = Input(shape=(config.MAX_GT_INSTANCES, 2), name='gt_class_ids')

[yizt]

@kaixinbaba 第一个是padding,第二个是分类，参数2 代表的是2个分类