odegeasslbc / FastGAN-pytorch

Official implementation of the paper "Towards Faster and Stabilized GAN Training for High-fidelity Few-shot Image Synthesis" in ICLR 2021
GNU General Public License v3.0
600 stars 100 forks source link

resolve Multi gpu #25

Closed Topology1225 closed 3 years ago

Topology1225 commented 3 years ago

close #8

Hello. I modified fastGAN so that it can be trained on multiple GPUs. Please run the following command to merge this pull request.

This pull request relates to the issue #8.

CUDA_VISIBLE_DEVICES=0,1 python  train.py --path ./few-shot-images/pokemon/img/ --batch_size 16