|
@@ -91,7 +91,7 @@ if __name__ == '__main__':
|
|
|
if args.model_file_filter:
|
|
|
onnx_files = [item for item in onnx_files if args.model_file_filter in item]
|
|
|
else:
|
|
|
- onnx_files = [item for item in onnx_files if "pruned" not in item]
|
|
|
+ onnx_files = [item for item in onnx_files]
|
|
|
print(f"model_name: {model_dir}\nonnx_files:")
|
|
|
print(*onnx_files, sep='\n')
|
|
|
for onnx_file in onnx_files:
|