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

update vae odel

parents dd78d680 27c4ad79
Branches
Tags
No related merge requests found
Pipeline #42696 failed
Stage: build
Stage: test
Stage: deploy
Show whitespace changes
Inline Side-by-side
video_prediction_savp/video_prediction/models/vanilla_vae_model.py
+ 2
− 1
View file @ 55749398
...@@ -74,6 +74,7 @@ class VanillaVAEVideoPredictionModel(BaseVideoPredictionModel): ...@@ -74,6 +74,7 @@ class VanillaVAEVideoPredictionModel(BaseVideoPredictionModel):
) )
return dict(itertools.chain(default_hparams.items(), hparams.items())) return dict(itertools.chain(default_hparams.items(), hparams.items()))
def build_graph(self,x): def build_graph(self,x):
self.x = x["images"] self.x = x["images"]
#self.global_step = tf.Variable(0, name = 'global_step', trainable = False) #self.global_step = tf.Variable(0, name = 'global_step', trainable = False)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment