Ошибка говорит о том, что типами для 2 последних элементов в вашем кортеже являются int и float, а все должно быть Tensor.
Вероятно, самый простой способ исправить это сделать int и float (mini_batch_seq_lengths
и annealing_factor
, я полагаю) Тензор: измените модель, чтобы принять Тензор, и при необходимости конвертируйте в / из скаляров.