latent-diffusion
latent-diffusion copied to clipboard
cannot run on multi gpus
I start training with this command
'python main.py --base configs/autoencoder/vqmodel1.yaml -t --gpus 4,5'
but I got this
everything works fine, steps in one epoch are halved, but only one gpu is in use, and only started one process.
How to solve this problem?
I start training with this command 'python main.py --base configs/autoencoder/vqmodel1.yaml -t --gpus 4,5' but I got this
everything works fine, steps in one epoch are halved, but only one gpu is in use, and only started one process. How to solve this problem?
have you found a solution yet? I am facing the same issue