search

cindyxinyiwang / deep-latent-sequence-model

Pytorch implementation of "A Probabilistic Formulation of Unsupervised Text Style Transfer" by He. et. al. at ICLR 2020
163 stars 26 forks source link

how to use multi-gpu for training #12

Closed NingHongKe closed 2 years ago

NingHongKe commented 3 years ago

I'm wondering how to use multi-gpu for training. export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7

image

jxhe commented 3 years ago

Sorry the code doesn't support multi-gpu training