mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Only fix weird path bug in windows

Browse Source
This commit is contained in:
James Betker 2021-08-05 22:21:25 -06:00
parent 70dcd1107f
commit 3ca51e80b2
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/trainer/networks.py
View File

@ -37,8 +37,9 @@ def find_registered_model_fns(base_path='models'):
found_fns = {} found_fns = {}
module_iter = pkgutil.walk_packages([base_path]) module_iter = pkgutil.walk_packages([base_path])
for mod in module_iter: for mod in module_iter:
if os.getcwd() not in mod.module_finder.path: if os.name == 'nt':
continue # I have no idea why this is necessary - I think it's a bug in the latest PyWindows release. if os.getcwd() not in mod.module_finder.path:
continue # I have no idea why this is necessary - I think it's a bug in the latest PyWindows release.
if mod.ispkg: if mod.ispkg:
EXCLUSION_LIST = ['flownet2'] EXCLUSION_LIST = ['flownet2']
if mod.name not in EXCLUSION_LIST: if mod.name not in EXCLUSION_LIST: