1. Learn
  2. /
  3. Courses
  4. /
  5. Deep Learning with PyTorch

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.