VQ-GAN could not use GPU for training

[Grople]

I trained the VQ-GAN, but the GPU are not used for acceleration. Could you tell me why? the training command is following: CUDA_VISIBLE_DEVICES=1 python main.py -b configs/brats_vqgan.yaml -t True --gpus 0

But the memory of gpu does not change.

[HBB0517]

hi, i want konw do you use seg data? original spade module seems need label (seg) data.