快递单信息抽取卡在了模型训练这一步，在线上运行是可以过的，但是在windows上运行时就报错了

model.fit(train_data=train_loader,
          eval_data=dev_loader,
          epochs=10,
          save_dir='./results',
          log_freq=1)

我查了一下，发现在网络构建这一步的forward函数中传参的类型发生了变化，线上运行时x和lens的dtype都是paddle.int64,但是在windows上就变成了paddle.int32,经过该函数处理后，lens的类型没变，返回参数pred的类型却是paddle.int64，这怎么处理啊，传参是在model里传过来的，我核对过版本号啊

    def forward(self, x, lens):
        print(x.dtype,lens.dtype)
        embs = self.word_emb(x)
        output, _ = self.gru(embs)
        output = self.fc(output)
        _, pred = self.decoder(output, lens)
        print(output.dtype,pred.dtype,lens.dtype)
        return output, lens, pred