conv2_fc2_cross_entropy_net.py 文件源码

def __init__(self, learning_rate, input_shape, BS):#input_shape example: [BS,1,28,28]
        self.lr = learning_rate

        self.conv2d_1 = ly.conv2d(input_shape, [5, 5, 1, 6], [2, 2], 'VALID')
        self.relu_1 = ly.relu()

        # conv2 : 6*12*12 - > 10*5*5
        self.conv2d_2 = ly.conv2d([BS, 6, 12, 12], [3, 3, 6, 10], [2, 2], 'VALID')
        self.relu_2 = ly.relu()

        self.flatter = ly.flatter()

        self.full_connect_1 = ly.full_connect(250, 84)
        self.relu_3 = ly.relu()

        self.full_connect_2 = ly.full_connect(84, 10)

        self.loss_func = ly.softmax_cross_entropy_error()