Skip to content

Instantly share code, notes, and snippets.

@securetorobert
Created October 1, 2020 16:13
Show Gist options
  • Save securetorobert/575c56254fbfef866d4446aee7fd5cfb to your computer and use it in GitHub Desktop.
Save securetorobert/575c56254fbfef866d4446aee7fd5cfb to your computer and use it in GitHub Desktop.
Model definition of AlexNet
model = models.Sequential([
layers.Conv2D(filters=96, kernel_size=(11, 11), strides=4, activation='relu', input_shape=(HEIGHT, WIDTH, NUM_CHANNELS), bias_initializer=tf.initializers.zeros(), kernel_initializer=tf.initializers.RandomNormal(mean=0, stddev=0.01)),
layers.BatchNormalization(),
layers.MaxPooling2D(),
layers.Conv2D(filters=256, kernel_size=(5, 5), activation='relu', bias_initializer=tf.initializers.ones(), kernel_initializer=tf.initializers.RandomNormal(mean=0, stddev=0.01)),
layers.BatchNormalization(),
layers.MaxPooling2D(),
layers.Conv2D(filters=384, kernel_size=(3, 3), activation='relu', bias_initializer=tf.initializers.zeros(), kernel_initializer=tf.initializers.RandomNormal(mean=0, stddev=0.01)),
layers.Conv2D(filters=384, kernel_size=(3, 3), activation='relu', bias_initializer=tf.initializers.ones(), kernel_initializer=tf.initializers.RandomNormal(mean=0, stddev=0.01)),
layers.Conv2D(filters=256, kernel_size=(3, 3), activation='relu', bias_initializer=tf.initializers.ones(), kernel_initializer=tf.initializers.RandomNormal(mean=0, stddev=0.01)),
layers.MaxPooling2D(),
layers.Dense(4096, activation='relu', bias_initializer=tf.initializers.ones(), kernel_initializer=tf.initializers.RandomNormal(mean=0, stddev=0.01)),
layers.Dropout(0.5),
layers.Dense(4096, activation='relu', bias_initializer=tf.initializers.ones(), kernel_initializer=tf.initializers.RandomNormal(mean=0, stddev=0.01)),
layers.Dropout(0.5),
layers.Dense(4096, activation='relu', kernel_initializer=tf.initializers.RandomNormal(mean=0, stddev=0.01)),
layers.Dense(NCLASSES, activation='softmax', bias_initializer=tf.initializers.ones())
])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment