Loading MNIST dataset...

Neural Network MNIST Model Trainer and Visualizer

Conv2D Layer

Kernel Size:
Filters:
Strides:

ReLU Activation

MaxPooling Layer

Pool Size:
Strides:

Conv2D Layer

Kernel Size:
Filters:
Strides:

ReLU Activation

MaxPooling Layer

Pool Size:
Strides:

Flatten Layer

Dense (Linear) Layer

Input Channels:
Output Channels:

Loss

Optimizer 
Learning Rate

Loss 
2nd Convolution layer weights are averaged during visualization for performance reasons. Try changing filter amounts if you wish to observe performance and time changes.