↓ 1 callersMethod__init__(self,
k,
width,
edge_bias,
conditional=False,
collection=None, name='coupling')
src/tf_components/bijector/Coupling.py:32
↓ 1 callersMethodsample_latent_mcmc(self,
conditioning=None,n_batch=1,temperature=1.0,
step_size=0.01,adaptation_steps=1,warmup_steps=3)
src/models/shared/Single_scale_flow.py:89
Method__init__(self,k_size,n_out,stride,edge_bias,activation='relu',
dtype=tf.float32,collection=None,name="openai_conv")
src/tf_components/openai_conv.py:12
Method__init__(self,k_size,n_out,stride,edge_bias,
dtype=tf.float32,collection=None,name="openai_zero_conv")
src/tf_components/openai_zero_conv.py:10
Method__init__(self,n_steps,width,edge_bias,conditional=False,collection=None,name='mutlistep',**remaining_kwargs)
src/tf_components/bijector/Multi_step.py:8
Method__init__(self,
width,
edge_bias,
conditional=False,
collection=None,name='step')
src/tf_components/bijector/Step.py:8
Method__init__(self,k_size,width,edge_bias,
dtype=tf.float32,collection=None,name="coupling_res_block")
src/tf_components/bijector/Coupling.py:9
Method__init__(self,n_steps,width,split,conditional=False,collection=None,name='mutlistep',**remaining_kwargs)
src/tf_components/bijector/Squeeze_step.py:7
Method__init__(self,shape,n_steps,coupling_width,spatial_bias,conditional=False,name='Norm_flow')
src/models/shared/Single_scale_flow.py:15