1. Learn
    2. /
  2. Courses
    3. /
  3. Introduction to Deep Learning with PyTorch

Exercise

Layer initialization

The initialization of the weights of a neural network has been the focus of researchers for many years. When training a network, the method used to initialize the weights has a direct impact on the final performance of the network.

As a machine learning practitioner, you should be able to experiment with different initialization strategies. In this exercise, you are creating a small neural network made of two layers and you are deciding to initialize each layer's weights with the uniform method.

Instructions

100 XP
  • For each layer, use the uniform initialization method to initialize the weights.