We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c72d6c9 commit 1c2a429Copy full SHA for 1c2a429
setup.py
@@ -32,7 +32,6 @@ def parse_requirements(fname):
32
33
34
def package_model_zoo():
35
- from ppdet.model_zoo import MODEL_ZOO_FILENAME
36
cur_dir = osp.dirname(osp.realpath(__file__))
37
cfg_dir = osp.join(cur_dir, "configs")
38
cfgs = glob.glob(osp.join(cfg_dir, '*/*.yml'))
@@ -44,7 +43,7 @@ def package_model_zoo():
44
43
valid_cfgs.append(cfg)
45
model_names = [osp.splitext(osp.split(cfg)[1])[0] for cfg in valid_cfgs]
46
47
- model_zoo_file = osp.join(cur_dir, 'ppdet', 'model_zoo', MODEL_ZOO_FILENAME)
+ model_zoo_file = osp.join(cur_dir, 'ppdet', 'model_zoo', 'MODEL_ZOO')
48
with open(model_zoo_file, 'w') as wf:
49
for model_name in model_names:
50
wf.write("{}\n".format(model_name))
0 commit comments