Skip to content
Snippets Groups Projects
Commit 55749398 authored by gong1's avatar gong1
Browse files

update vae odel

parents dd78d680 27c4ad79
No related branches found
No related tags found
No related merge requests found
Pipeline #42696 failed
......@@ -74,6 +74,7 @@ class VanillaVAEVideoPredictionModel(BaseVideoPredictionModel):
)
return dict(itertools.chain(default_hparams.items(), hparams.items()))
def build_graph(self,x):
self.x = x["images"]
#self.global_step = tf.Variable(0, name = 'global_step', trainable = False)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment