From 9f079c65054f86f8df65678e58e7692c869dd073 Mon Sep 17 00:00:00 2001 From: Xiaoyang Guo Date: Sun, 14 Apr 2019 09:28:06 -0400 Subject: [PATCH] add math package back --- models/gwcnet.py | 1 + 1 file changed, 1 insertion(+) diff --git a/models/gwcnet.py b/models/gwcnet.py index 4746807..adc4f29 100644 --- 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):