Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
config.yml 380 Bytes
Newer Older
colorjam's avatar
Refactor model compression examples (#3326)  
colorjam committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 
searchSpace:
  sparsity:
    _type: choice
    _value: [0.25, 0.5, 0.75]
  pruner:
    _type: choice
    _value: ['slim', 'l2filter', 'fpgm', 'apoz']
  model:
    _type: choice
    _value: ['vgg16', 'vgg19']
trainingService:
  platform: local
trialCodeDirectory: .
trialCommand: python3 basic_pruners_torch.py --nni
trialConcurrency: 1
trialGpuNumber: 0
tuner:
BUPTLYH's avatar
modify the config bug about Automatic Model Pruning using NNI Tuners (#3541)  
BUPTLYH committed
18 
  name: GridSearch