Update stylex_train.py to fix error when `sample_from_encoder` is False

NoahVl / Explaining-In-Style-Reproducibility-Study

Re-implementation of the StylEx paper, training a GAN to explain a classifier in StyleSpace, paper by Lang et al. (2021).

Other

36 stars 8 forks source link

Closed tmabraham closed 2 years ago

tmabraham commented 2 years ago

I get an error when sample_from_encoder is False, this solves that error.

NoahVl commented 2 years ago

Ah yes, we always tested with encoder_input=True to test how well the reconstruction was going.

Thanks for the fix!