Skip to content

Commit

Permalink
updated
Browse files Browse the repository at this point in the history
  • Loading branch information
haowen-xu committed Feb 22, 2020
1 parent bd41bef commit c67f101
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions tensorkit/examples/auto_encoders/vae_realnvp_posterior.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

class Config(mltk.Config):
# model parameters
z_dim: int = 4
z_dim: int = 40
flow_levels: int = 10
flow_hidden_layer_count: int = 1
flow_hidden_layer_units: int = 250
Expand Down Expand Up @@ -71,8 +71,7 @@ def __init__(self, x_dim: int, config: Config):
],
shared=b.build(),
)
flows.append(tk.flows.CouplingLayer(
shift_and_pre_scale, scale='sigmoid'))
flows.append(tk.flows.CouplingLayer(shift_and_pre_scale, scale='sigmoid'))

# feature rearrangement by invertible dense
flows.append(tk.flows.InvertibleDense(config.z_dim))
Expand Down

0 comments on commit c67f101

Please sign in to comment.