|
@@ -28,7 +28,7 @@ def visualize(img_file,
|
|
|
num_samples=3000,
|
|
num_samples=3000,
|
|
|
batch_size=50,
|
|
batch_size=50,
|
|
|
save_dir='./'):
|
|
save_dir='./'):
|
|
|
- model_info = self.get_model_info()
|
|
|
|
|
|
|
+ model_info = moedl.get_model_info()
|
|
|
if model_info['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(
|