About using average weights

[xyimaging]

Thanks for your good work!

In the s_ugatit_plus.py, you added a using_avg_w_forward function but didn't use it. I'm curious if you have any ablation studies or insight about this? Will generating w via averaging z from multi-layer MLP leads to a better or more stable result? Thanks!

[xyimaging]

correct a mistake: it should be multiple Z from the same MLP, not Z from different layers of MLP