Skip to content

Instantly share code, notes, and snippets.

@emilwallner
Created September 24, 2017 12:37
Show Gist options
  • Save emilwallner/fa18a6ab939933219c9b4efb8e50fa58 to your computer and use it in GitHub Desktop.
Save emilwallner/fa18a6ab939933219c9b4efb8e50fa58 to your computer and use it in GitHub Desktop.
network = input_data(shape=[None, 32, 32, 3],
data_preprocessing=img_prep,
data_augmentation=img_aug)
network = conv_2d(network, 32, 3, activation='relu', regularizer='L2')
network = max_pool_2d(network, 2)
network = conv_2d(network, 64, 3, activation='relu', regularizer='L2')
network = conv_2d(network, 64, 3, activation='relu', regularizer='L2')
network = max_pool_2d(network, 2)
network = fully_connected(network, 512, activation='relu', regularizer='L2')
network = fully_connected(network, 10, activation='softmax')
network = regression(network, optimizer='adam',
loss='categorical_crossentropy',
learning_rate=0.001)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment