Skip to content

Instantly share code, notes, and snippets.

Avatar

Patrick Haluptzok haluptzok

View GitHub Profile
@haluptzok
haluptzok / pg-pong.py
Last active August 9, 2022 15:55 — forked from karpathy/pg-pong.py
Training a Neural Network ATARI Pong agent with Policy Gradients from raw pixels
View pg-pong.py
""" Trains an agent with (stochastic) Policy Gradients on Pong. Uses OpenAI Gym. """
import numpy as np
import pickle
import gym
# hyperparameters
H = 200 # number of hidden layer neurons
batch_size = 10 # every how many episodes to do a param update?
learning_rate = 1e-4
gamma = 0.99 # discount factor for reward