Fix import

This commit is contained in:
Shangtong Zhang
2018-05-07 21:24:52 -06:00
parent fb95820803
commit fa294cd532
2 changed files with 3 additions and 2 deletions
+1
View File
@@ -7,6 +7,7 @@
import torch
import torch.nn as nn
import torch.nn.functional as F
import numpy as np
class BaseNet:
def set_gpu(self, gpu):