def get_gen_variables(self):
weights = {
'wd': tf.Variable(tf.random_normal([self.sample_size+self.y_dim , 1024], stddev=0.02), name='gen_w1'),
'wc1': tf.Variable(tf.random_normal([1024 + self.y_dim , 7 * 7 * 128], stddev=0.02), name='gen_w2'),
'wc2': tf.Variable(tf.random_normal([4, 4, 64, 128 + self.y_dim], stddev=0.02), name='gen_w3'),
'wc3': tf.Variable(tf.random_normal([4, 4, 1, 64 + self.y_dim], stddev=0.02), name='gen_w4'),
}
biases = {
'bd': tf.Variable(tf.zeros([1024]), name='gen_b1'),
'bc1': tf.Variable(tf.zeros([7 * 7 * 128]), name='gen_b2'),
'bc2': tf.Variable(tf.zeros([64]), name='gen_b3'),
'bc3': tf.Variable(tf.zeros([1]), name='gen_b4')
}
return weights, biases
评论列表
文章目录