search

NVlabs / NVAE

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
https://arxiv.org/abs/2007.03898
Other
1.02k stars 164 forks source link

"arch_instance" argument #48

Open SerezD opened 11 months ago

SerezD commented 11 months ago

Hi there, thank you for the great job!

I noticed that your implementation has an argument called arch_instance, set to default as res_mbconv. From the provided readme, it appears that this parameter is set as default in all your runs.

My question is: is this just an old argument used for testing different architectures? When training a new model, is it correct to leave it as default? Can you also please provide some more info on what this argument does/what are the different tried architectures?

Thank you