|
|
@@ -28,8 +28,7 @@ def visualize(img_file,
|
|
|
num_samples=3000,
|
|
|
batch_size=50,
|
|
|
save_dir='./'):
|
|
|
- model_info = moedl.get_model_info()
|
|
|
- if model_info['status'] != 'Normal':
|
|
|
+ if model.status != 'Normal':
|
|
|
raise Exception('The explanation only can deal with the Normal model')
|
|
|
model.arrange_transforms(
|
|
|
transforms=model.test_transforms, mode='test')
|