Changing dimension of z_sem

[yair-schiff]

Thanks for open-sourcing this repo!

I wanted to confirm which config parameters need to be changed in order to test different sizes for z_sem in the CelebA experiment. Am I right in assuming I need to change the following 3 lines?

1. TrainConfig -> net_beatgans_embed_channels

    net_beatgans_embed_channels: int = 512

2. TrainConfig -> style_ch

    style_ch: int = 512

3. autoenc_base -> conf.net_beatgans_embed_channels

        conf.net_beatgans_embed_channels = 512

Am I missing any other locations that need to be changed (e.g., for ddpm) or am I changing too many config parameters?

When I tried changing only TrainConfig -> net_beatgans_embed_channels I ran into tensor multiplication dim errors for training the latent DDIM.

[konpatp]

I myself am not sure about all the places that needed to change. Sorry about that. However, you don't need to change 1) because it is just a default value which would definitely be replaced by the actual settings from 3). I would suggest that you duplicate the function autoenc_base and then just override the specified parameter, just like 3). For 2), I'm not sure for now, may need to do dig into the dependencies to make sure.