Exercise

Your first PyTorch neural network

You are going to build the same neural network you built in the previous exercise, but now using the PyTorch way. As a reminder, you have 784 units in the input layer, 200 hidden units and 10 units for the output layer.

Instructions

100 XP
  • Instantiate two linear layers calling them self.fc1 and self.fc2. Determine their correct dimensions.
  • Implement the .forward() method, using the two layers you defined and returning x.