使用hub的erine模型但是trainer.train（）的时候出现问题
收藏
trainer.train(
train_dataset,
epochs=10,
batch_size=32,
save_interval=1,)
代码前几天运行还没问题，今天运行出现了以下错误ValueError: label should not out of bound, but gotTensor(shape=[1], dtype=int64, place=CUDAPlace(0), stop_gradient=True,
[2]
0
收藏
请登录后评论
看了相似的帖子，说标签的问题，但是我标签只有3类，定义的时候是没问题啊
标签定义
数据标签没问题也，但测试集标签全为0却没问题