import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
class BNMnist:
def __init__(self):
print ("initialize......")
self.epoches = 1000
self.batch_size = 100
self.mnist = input_data.read_data_sets('MNIST_DATA/',one_hot=True)
self.learning_rate = 0.001
self.regularizer_rate = 0.00001
def Train(self):
print ("Train......")
weights = {
"W1":tf.Variable(tf.random_normal(shape=[784,512],dtype=tf.float32,stddev=0.01)),
"W2":tf.Variable(tf.random_normal(shape=[512,256],dtype=tf.float32,stddev=0.01)),
"Output":tf.Variable(tf.random_normal(shape=[256,10],dtype=tf.float32,stddev=0.01))
}
bias = {
'b1':tf.Variable(tf.zeros(shape=[512],dtype=tf.float32)),
Tensorflow with Batch normalization
最新推荐文章于 2021-05-28 11:09:40 发布