Update transformation.py

2020-01-17 00:12:18 +09:00 · 2020-01-17 00:12:18 +09:00 · 78d7ed96a9
parent 382a9b50f3
commit 78d7ed96a9
1 changed files with 1 additions and 0 deletions
--- 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 """