add math package back

2019-04-14 09:28:06 -04:00 · 2019-04-14 09:28:06 -04:00 · 9f079c6505
commit 9f079c6505
parent 1167dcb794
1 changed files with 1 additions and 0 deletions
--- a/models/gwcnet.py
+++ b/models/gwcnet.py
@ -5,6 +5,7 @@ import torch.utils.data
 from torch.autograd import Variable
 import torch.nn.functional as F
 from models.submodule import *
+import math


 class feature_extraction(nn.Module):