Yuqi Wu

icon indicating a website link www.bilibili.com icon indicating an email [email protected]

icon company Tsinghua University icon location Beijing, China icon location Wu Yuqi, a student in Tsinghua University.Study in Github and try my best to make some contributions to this community.

Results 1 issues of Yuqi Wu

Cannot use gpu to fine-tune on custom-data

2 comment

I used "python main.py --base configs/custom_vqgan.yaml -t True --gpus 0" to fine-tune a pretrained model on my own data, but it seems that **it always train on my cpu, not...