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

address/correct the number of samples per epoch number issue

parent c689e922
No related branches found
No related tags found
No related merge requests found
Pipeline #42516 failed
......@@ -357,6 +357,7 @@ def main():
sess.graph.as_default()
sess.run(tf.global_variables_initializer())
sess.run(tf.local_variables_initializer())
model.restore(sess, args.checkpoint)
#model.restore(sess, args.checkpoint)#Bing: Todo: 20200728 Let's only focus on true and persistend data
sample_ind, gen_images_all, persistent_images_all, input_images_all = initia_save_data()
......
......@@ -366,13 +366,13 @@ def main():
# "2012":[1,2,3,4,5,6,7,8,9,10,11,12],
# "2013_complete":[1,2,3,4,5,6,7,8,9,10,11,12],
# "2015":[1,2,3,4,5,6,7,8,9,10,11,12],
"2017":[1,2,3,4,5,6,7,8,9,10]
"2017_test":[1,2,3,4,5,6,7,8,9,10]
},
"val":
{"2017":[11]
{"2017_test":[11]
},
"test":
{"2017":[12]
{"2017_test":[12]
}
}
......
......@@ -41,7 +41,7 @@ class VanillaConvLstmVideoPredictionModel(BaseVideoPredictionModel):
lr: learning rate. if decay steps is non-zero, this is the
learning rate for steps <= decay_step.
max_steps: number of training steps.
context_frames: the number of ground-truth frames to pass in at
context_frames: the number of ground-truth frames to pass :qin at
start. Must be specified during instantiation.
sequence_length: the number of frames in the video sequence,
including the context frames, so this model predicts
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment