Head: The head consists of two 3x3 convolution layers, with each layer being followed by a batch normalization layer[1]} and a leaky relu[2]} non linearity. The first layer has a stride of 2 and 16 filters. The second layer has 64 filters.
