Налаштування середовища та нейронної мережі
Почнете з налаштування середовища, яке ви використовуватимете протягом курсу: середовища 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)