提交 b1797a34 编写于 作者: M Mark Ma

fixed InceptionV3 dygraph class creation is not wrapped with fluid.dygraph.guard()

上级 35eb66c3
......@@ -250,7 +250,8 @@ def calculate_fid_given_paths(paths,
if not os.path.exists(p):
raise RuntimeError('Invalid path: %s' % p)
if model is None:
if model is None and style != 'stargan':
with fluid.dygraph.guard():
block_idx = InceptionV3.BLOCK_INDEX_BY_DIM[dims]
model = InceptionV3([block_idx], class_dim=1008)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册