"""Training script for VQGAN."""