Using the provided checkpoints (mentioned in README), we cannot reproduce the same results reported. Can you provide the exact configs ?