Please clarify Input and Output node names for trained model - model_250epochs_ft_Final/model_250.ckpt.meta
rkartheek2022 opened this issue · 0 comments
Hi All,
What would be the input node name and output node name for the below model?
model_250epochs_ft_Final/model_250.ckpt.meta
From the code https://github.com/daigo0927/pwcnet/blob/master/train.py
Looks like the input node name is 'images' and output node name is 'flows'. How ever - SNPE tools errors on input node.
Coudl some one please confirm the input node name for model_250epochs_ft_Final/model_250.ckpt.meta model?
with tf.name_scope('Data'):
self.images = tf.placeholder(tf.float32, shape = (self.args.batch_size, 2, *self.image_size, 3),
name = 'images')
images_0, images_1 = tf.unstack(self.images, axis = 1)
self.flows_gt = tf.placeholder(tf.float32, shape = (self.args.batch_size, *self.image_size, 2),
name = 'flows')
Thanks,
Rathi