CommencezCommencez gratuitement

Configuration de l'environnement et du réseau de neurones

Vous allez commencer par configurer l'environnement que vous utiliserez tout au long du cours : l'environnement Lunar Lander, où un agent contrôle les propulseurs d'un véhicule qui tente d'atterrir sur la Lune.

torch, torch.nn, torch.optim et gym sont importés dans vos exercices.

Cette activité fait partie du cours

Deep Reinforcement Learning en Python

Voir le cours

Instructions de l’exercice

  • Initialisez l'environnement Lunar Lander dans gym (LunarLander-v2).
  • Définissez une seule couche de transformation linéaire, avec une dimension d'entrée dim_inputs et une dimension de sortie dim_outputs.
  • Instanciez le réseau de neurones avec une dimension d'entrée 8 et une dimension de sortie 4.
  • Fournissez les paramètres à l'optimiseur Adam.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Initiate the Lunar Lander environment
env = gym.____

class Network(nn.Module):
    def __init__(self, dim_inputs, dim_outputs):
        super(Network, self).__init__()
        # Define a linear transformation layer 
        self.linear = ____
    def forward(self, x):
        return self.linear(x)

# Instantiate the network
network = ____

# Initialize the optimizer
optimizer = optim.Adam(____, lr=0.0001)

print("Network initialized as:\n", network)
Modifier et exécuter le code