在模型训练过程中遇到问题

EZ3002: Optype [TransData] of Ops kernel [AIcoreEngine] is unsupported. Reason: [Dynamic shape check]: The format and dtype is not precisely equivalent to format and dtype in op information library[Static shape check]:The format and dtype is not precisely equivalent to format and dtype in op information library[tbe-custom]:op type TransData is not found in this op store.[tbe-builtin]:[Dynamic shape check]: The format and dtype is not precisely equivalent to format and dtype in op information library[Static shape check]:The format and dtype is not precisely equivalent to format and dtype in op information library. 

此问题是模型训练第二个batch时，加载输入数据到npu上出现的

即更新模型参数后，执行

batch_x = batch_x.to("npu:0")
batch_y = batch_y.to("npu:0")

出现如上报错。

经过测试，模型推理时并无报错。