Hi, interesting work! Can you share the approximate time to train the VQVAE and the task solver on both tasks? Thanks!