未验证 提交 7e661ba9 编写于 作者: S SunAhong1993 提交者: GitHub

Update visualize.py

上级 3b2feefb
...@@ -28,8 +28,7 @@ def visualize(img_file, ...@@ -28,8 +28,7 @@ def visualize(img_file,
num_samples=3000, num_samples=3000,
batch_size=50, batch_size=50,
save_dir='./'): save_dir='./'):
model_info = moedl.get_model_info() if model.status != 'Normal':
if model_info['status'] != 'Normal':
raise Exception('The explanation only can deal with the Normal model') raise Exception('The explanation only can deal with the Normal model')
model.arrange_transforms( model.arrange_transforms(
transforms=model.test_transforms, mode='test') transforms=model.test_transforms, mode='test')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册