Alexander Penkin SSS135
SSS135
/ batch_renormalization.py
Last active
June 4, 2018 20:54
Testing Batch Renormalization with Tensor Comprehensions
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| 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 |