НачатьНачать бесплатно

Настройка среды и нейронной сети

Начнём с настройки среды, которую вы будете использовать на протяжении всего курса: это среда Lunar Lander, где агент управляет двигателями аппарата, пытающегося приземлиться на Луне.

Библиотеки torch, torch.nn, torch.optim и gym уже импортированы в упражнения.

Это упражнение является частью курса

Глубокое обучение с подкреплением на Python

Посмотреть курс

Инструкции к упражнению

  • Инициализируйте среду Lunar Lander в gym (LunarLander-v2).
  • Определите один слой линейного преобразования с входным измерением dim_inputs и выходным измерением dim_outputs.
  • Создайте экземпляр нейронной сети с входным измерением 8 и выходным измерением 4.
  • Передайте оптимизатору Adam параметры сети.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# 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)
Редактировать и запускать код