Pytorch 中“Assertion `t ＞= 0 && t ＜ n_classes` failed error ” 解决方案

一、问题描述

关键错误为：

ClassNLLCriterion_updateOutput_no_reduce_kernel: block: [0,0,0], thread: [1,0,0] Assertion `cur_target >= 0 && cur_target < n_classes` failed.

二、问题排查

1、训练使用的model是transformers.OpenAIGPTLMHeadModel，对应的参数：