未验证 提交 74e2185f 编写于 作者: 走神的阿圆's avatar 走神的阿圆 提交者: GitHub

fix bug of embedding (#935)

Co-authored-by: NPeter Pan <littlepanzh@gmail.com>
上级 8d4e7fa9
...@@ -271,6 +271,7 @@ def get_embeddings_list(log_reader): ...@@ -271,6 +271,7 @@ def get_embeddings_list(log_reader):
run2tag = get_logs(log_reader, 'embeddings') run2tag = get_logs(log_reader, 'embeddings')
for run, _tags in zip(run2tag['runs'], run2tag['tags']): for run, _tags in zip(run2tag['runs'], run2tag['tags']):
run = log_reader.name2tags[run] if run in log_reader.name2tags else run
for tag in _tags: for tag in _tags:
name = path = os.path.join(run, tag) name = path = os.path.join(run, tag)
if name in EMBEDDING_NAME: if name in EMBEDDING_NAME:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册