update

alibaba · Apr 10, 2024 · 301475f · 301475f
1 parent f0222e5
commit 301475f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/configs/classification/imagenet/resnet/imagenet_resnet50_jpg.py b/configs/classification/imagenet/resnet/imagenet_resnet50_jpg.py
@@ -6,3 +6,6 @@
         depth=50,
         out_indices=[4],  # 0: conv-1, x: stage-x
         norm_cfg=dict(type='BN')))
+
+export_type = 'onnx'
+export = dict(export_type=export_type)