Alexander Penkin SSS135

Unity / Machine Learning programmer. See my PyTorch Reinforcement Learning framework SSS135/pytorch-rl-kit or Unity 2D lighting plugin SSS135/Light2D

SSS135 / batch_renormalization.py

Last active June 4, 2018 20:54

Testing Batch Renormalization with Tensor Comprehensions

	import time

	import tensor_comprehensions as tc
	import tensor_comprehensions.tc_unit as tcu
	import torch
	import torch.cuda
	import torch.nn as nn
	import torch.nn.functional as F
	from torch.autograd import Variable, Function