add torch train to vae#44

letiziaia wants to merge 4 commits intoAlexanderVNikitin:mainAlexanderVNikitin/tsgm:mainfrom

letiziaia:Feature/torch-betavaeletiziaia/tsgm:Feature/torch-betavaeCopy head branch name to clipboard