Ver código fonte

add prune configs and prompt

sunyanfang01 5 anos atrás
pai
commit
07f9ad9095
1 arquivos alterados com 2 adições e 0 exclusões
  1. 2 0
      paddlex/cv/models/slim/prune_config.py

+ 2 - 0
paddlex/cv/models/slim/prune_config.py

@@ -57,6 +57,8 @@ sensitivities_data = {
     'https://bj.bcebos.com/paddlex/slim_prune/xception41.sensitivities',
     'Xception65':
     'https://bj.bcebos.com/paddlex/slim_prune/xception65.sensitivities',
+    'ShuffleNetV2':
+    'https://bj.bcebos.com/paddlex/slim_prune/shufflenetv2_sensitivities.data',
     'YOLOv3_MobileNetV1':
     'https://bj.bcebos.com/paddlex/slim_prune/yolov3_mobilenetv1.sensitivities',
     'YOLOv3_MobileNetV3_large':