Update modules.py

qubvel · web-flow · commit 985ad7158144 · 2020-02-26T15:47:13.000+03:00
diff --git a/segmentation_models_pytorch/base/modules.py b/segmentation_models_pytorch/base/modules.py
@@ -1,3 +1,4 @@
+import torch
 import torch.nn as nn
 
 try:
