Hi, great work! Would you mind sharing the training script? It’d be super helpful for learning and reproducing the results. Thanks!