model_me.py 文件源码

def __init__(self,
               vocabulary_size=1295,
               mparams=None):

    super(Model, self).__init__()
    self._params = ModelParams(vocabulary_size=vocabulary_size)
    self.learning_rate = 1e-5

    self.images = tf.placeholder(dtype=tf.float32, shape=[None, None, 224, 224, 3], name='images')
    self.seqs_length = tf.placeholder(dtype=tf.int32, shape=[None], name='seqs_length')
    self.targets = tf.sparse_placeholder(tf.int32, name='targets')

    self._build_training()

    # # Saver
    # with tf.device('/cpu:0'):
    #   self.saver = tf.train.Saver(tf.all_variables())