def get_dis_variables(self):
weights = {
'wc1': tf.Variable(tf.random_normal([4, 4, 1 + self.y_dim, 64], stddev=0.02), name='dis_w1'),
'wc2': tf.Variable(tf.random_normal([4, 4, 64 + self.y_dim, 128], stddev=0.02), name='dis_w2'),
'wc3': tf.Variable(tf.random_normal([128 * 7 * 7 + self.y_dim, 1024], stddev=0.02), name='dis_w3'),
'wd': tf.Variable(tf.random_normal([1024 + self.y_dim, 1], stddev=0.02), name='dis_w4')
}
biases = {
'bc1': tf.Variable(tf.zeros([64]), name='dis_b1'),
'bc2': tf.Variable(tf.zeros([128]), name='dis_b2'),
'bc3': tf.Variable(tf.zeros([1024]), name='dis_b3'),
'bd': tf.Variable(tf.zeros([1]), name='dis_b4')
}
return weights, biases
评论列表
文章目录