Make Your First GAN Using PyTorch/

...

/

🍀 Challenge: Define the Neural Network Architecture

🍀 Challenge: Define the Neural Network Architecture

In this challenge, you are required to define the neural network architecture, the loss function, and the optimiser for the FMNIST data classification.

We'll cover the following...

Problem statement

In this challenge, your task is to define the:

  • Network architecture

    • The input layer should have 784 nodes.
    • The hidden layer should have 200 nodes with the LeakyReLU
...