From 78d7ed96a93c7472e50480437731f9f37c7b5727 Mon Sep 17 00:00:00 2001 From: Baek JeongHun Date: Fri, 17 Jan 2020 00:12:18 +0900 Subject: [PATCH] Update transformation.py --- modules/transformation.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/transformation.py b/modules/transformation.py index c43f173..311838a 100755 --- a/modules/transformation.py +++ b/modules/transformation.py @@ -4,6 +4,7 @@ import torch.nn as nn import torch.nn.functional as F device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') + class TPS_SpatialTransformerNetwork(nn.Module): """ Rectification Network of RARE, namely TPS based STN """