The first method we will define for our SimpleCNN class is the build method, which is responsible for building the architecture of our CNN. Our build method takes two pieces of input: the input tensor and the number of classes it should expect:
def build(self, input_tensor, num_classes): """ Builds a convolutional neural network according to the input shape and the number of classes. Architecture is fixed. Args: input_tensor: Tensor of the input num_classes: (int) number of classes Returns: The output logits before softmax """
We will first initialize tf.placeholder, called is_training. TensorFlow placeholders are like variables that don't have values. We only pass them values when we actually train the network and call the relevant ...
