Parag Jain parajain

PhD student at the University of Edinburgh

parajain / gradient_scaling.py

Last active November 2, 2017 14:36

Trying rl byscaling gradients with rewards

	import numpy as np
	import cPickle as pickle
	import gym
	import torch
	import torch.nn as nn
	from torch import optim
	from torch.autograd import Variable
	from torch import optim
	import torch.nn.functional as F