Update KERMIT.py

umbylyno · May 2, 2022 · a418e54 · a418e54
1 parent 1018324
commit a418e54
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/models/GraphBased/KERMIT.py b/models/GraphBased/KERMIT.py
@@ -1,3 +1,6 @@
+import torch, torchvision
+from torch import nn
+
 class ImageEncoder(nn.Module):
     def __init__(self, **kwargs):
         super(ImageEncoder, self).__init__()