Update transformation.py

zihaomu · Jan 16, 2020 · 78d7ed9 · 78d7ed9
1 parent 382a9b5
commit 78d7ed9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/modules/transformation.py b/modules/transformation.py
@@ -4,6 +4,7 @@
 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 """