From e23fd2a57035f3dccab5b9ca3450c02987d4b30a Mon Sep 17 00:00:00 2001 From: Andre Anjos <andre.dos.anjos@gmail.com> Date: Thu, 9 Apr 2020 08:12:11 +0200 Subject: [PATCH] [engine.predictor] Import fix --- bob/ip/binseg/engine/predictor.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/bob/ip/binseg/engine/predictor.py b/bob/ip/binseg/engine/predictor.py index 0a63d74a..62446d15 100644 --- a/bob/ip/binseg/engine/predictor.py +++ b/bob/ip/binseg/engine/predictor.py @@ -14,6 +14,8 @@ import torchvision.transforms.functional as VF import bob.io.base +from ..utils.summary import summary + import logging logger = logging.getLogger(__name__) -- GitLab