ComeçarComece de graça

Inicialização de camadas

A inicialização dos pesos de uma rede neural tem sido o foco dos pesquisadores há muitos anos. Ao treinar uma rede, o método usado para inicializar os pesos tem um impacto direto no desempenho final da rede.

Como praticante de aprendizado de máquina, você deve ser capaz de testar diferentes estratégias de inicialização. Neste exercício, você está criando uma pequena rede neural composta de duas camadas e decide inicializar os pesos de cada camada com o método uniforme.

Este exercício faz parte do curso

Introdução ao Aprendizado Profundo com o PyTorch

Ver curso

Instruções do exercício

  • Para cada camada (layer0 e layer1), use o método de inicialização uniforme para inicializar os pesos.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

layer0 = nn.Linear(16, 32)
layer1 = nn.Linear(32, 64)

# Use uniform initialization for layer0 and layer1 weights
nn.____.____(____)
nn.____.____(____)

model = nn.Sequential(layer0, layer1)
Editar e executar o código