Hi!
I’m implementing Normalizing Flows for zero-shot learning.
I have a quick question about Distributions — Pyro documentation. What is the purpose of param_dims in defining the hypernet?
Thanks!
Hi!
I’m implementing Normalizing Flows for zero-shot learning.
I have a quick question about Distributions — Pyro documentation. What is the purpose of param_dims in defining the hypernet?
Thanks!
param_dims
defines the output shape of the conditioning network. For the AffineCoupling
transform it has a single mean
and log(variance)
for each dimension that is modelled, and there are input_dim-split_dim
dimensions