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)