add math package back

This commit is contained in:
Xiaoyang Guo 2019-04-14 09:28:06 -04:00
parent 1167dcb794
commit 9f079c6505

View File

@ -5,6 +5,7 @@ import torch.utils.data
from torch.autograd import Variable from torch.autograd import Variable
import torch.nn.functional as F import torch.nn.functional as F
from models.submodule import * from models.submodule import *
import math
class feature_extraction(nn.Module): class feature_extraction(nn.Module):