Fix lr_schedule update for multi-gpu settings on train_vae#15

miquel-espinosa wants to merge 1 commit intoGuHuangAI:mainGuHuangAI/ADM-Public:mainfrom

miquel-espinosa:patch-1miquel-espinosa/ADM-Public:patch-1Copy head branch name to clipboard