Fix import bug re: Issue #13.

krasserm · Jul 3, 2020 · 6ab03a7 · 6ab03a7
1 parent 92990bf
commit 6ab03a7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/model/inception.py b/model/inception.py
@@ -1,4 +1,5 @@
 import torch
+import torch.nn.functional as F
 
 from torchvision import models
 from torchvision.models.utils import load_state_dict_from_url