###name: Age Classification CNN
caffemodel: age_net.caffemodel
caffemodel_url: https://dl.dropboxusercontent.com/u/38822310/age_net.caffemodel
""" Trains an agent with (stochastic) Policy Gradients on Pong. Uses OpenAI Gym. """ | |
import numpy as np | |
import cPickle as 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 |
###name: Age Classification CNN
caffemodel: age_net.caffemodel
caffemodel_url: https://dl.dropboxusercontent.com/u/38822310/age_net.caffemodel