Open ahmad-573 opened 1 month ago
I am trying to train the CausalVAE on my own dataset on 4 gpus but all memory is just used by device 0. Is distributed processing not incorporated into the training code?
set CUDA_VISABLE_DEVICES in training script.
I am trying to train the CausalVAE on my own dataset on 4 gpus but all memory is just used by device 0. Is distributed processing not incorporated into the training code?