Note that we have updated the pretrained transfuser model with the checkpoint submitted to the leaderboard. This model is trained on a different dataset than the one provided in this repository. (We are currently unable to share the entire dataset due to some issues.)