style-transformer icon indicating copy to clipboard operation
style-transformer copied to clipboard

Potential bug in train.py

Open xiaomaiaa opened this issue 2 years ago • 0 comments

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?

xiaomaiaa avatar Jun 27 '22 06:06 xiaomaiaa