Impressive work! Could you please release the training code for Stage 1 of the VGGAN codebook learning?