新手菜鸟级别,大神请绕道
今天较好的完成了cs231n-2017-assignments2-TensorFlow.ipynb中的最后一个部分:构建一个model对cifar数据集进行训练,今天的精度在train上达到0.98,在val上达到0.82,明天继续进行优化。
具体代码见TensorFlow.ipynb
以下仅贴出model:
def my_model(X,y,is_training):
# [conv-relu-conv-relu-pool] out=14x14
conv1 = tf.layers.conv2d(X,128,kernel_size=[3,3],strides=(1,1),activation=tf.nn.relu)
ba1 = tf.layers.batch_normalization(conv1,training=is_training)
conv2 = tf.layers.conv2d(ba1,256,[3,3],activation=tf.nn.relu)
ba2 = tf.layers.batch_normalization(conv2,training=is_training)
pool1 = tf.layers.max_