From 301475fe50ed4d4c0e748629fa1dd0ae2269a63d Mon Sep 17 00:00:00 2001 From: lxs01517514 Date: Wed, 10 Apr 2024 18:50:39 +0800 Subject: [PATCH] update --- .../classification/imagenet/resnet/imagenet_resnet50_jpg.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/configs/classification/imagenet/resnet/imagenet_resnet50_jpg.py b/configs/classification/imagenet/resnet/imagenet_resnet50_jpg.py index 99f2d169..29f5df1d 100644 --- 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)