From c51bf437bb144a40271def359a0d659aacfc5816 Mon Sep 17 00:00:00 2001 From: kremnik Date: Tue, 22 Sep 2020 16:30:36 +0300 Subject: [PATCH] resolve ModuleNotFound problem --- bin/infer_on_folders.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/bin/infer_on_folders.py b/bin/infer_on_folders.py index e9d5f10..c44bd31 100644 --- a/bin/infer_on_folders.py +++ b/bin/infer_on_folders.py @@ -1,6 +1,8 @@ import argparse import glob import os +import sys +sys.path.append('.') import torch from torchvision import transforms