YunZhu

icon company Nanjing University of Science and Technology icon location Nanjing, China

Results 1 comments of YunZhu

IndexError: The shape of the mask [6, 440] at index 1 does not match the shape of the indexed tensor [6, 760, 256] at index 1

I found that change the code truncate_len = tokenizer.model_max_length - 577 to truncate_len = tokenizer.model_max_length - 255 will solve it, but I don't know why it was set to 577.