Potential bug in train.py #14
Description
The code in Line 205 in file train.py "adv_log_porbs = model_D(gen_soft_tokens, gen_lengths, rev_styles)" should be in the ELSE? Because it uses style information in classfication and should be "Cond" method?