[DQN-1] Coder le réseau MLP dans agent.py

[Celz-Pch]

Objectif

Implémentation du réseau de neurones dans agent.py.

Architecture

Input (8) → Linear(256) → ReLU → Linear(256) → ReLU → Output(4)

Tasks

• Créer la classe QNetwork(nn.Module) avec PyTorch
• Méthode forward(state) → Q-values pour les 4 actions
• Tester que le réseau accepte un batch de states (batch_size, 8)

Critère de succès

net = QNetwork(state_dim=8, action_dim=4)
out = net(torch.randn(64, 8))
assert out.shape == (64, 4)