File size: 556 Bytes
419e1f1
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
!LayerPruningModifier
    end_epoch: -1.0
    layers: ['bert.encoder.layer.3', 'bert.encoder.layer.4', 'bert.encoder.layer.5', 'bert.encoder.layer.6', 'bert.encoder.layer.7', 'bert.encoder.layer.8', 'bert.encoder.layer.9', 'bert.encoder.layer.10', 'bert.encoder.layer.11']
    start_epoch: -1.0
    update_frequency: -1.0

!QuantizationModifier
    disable_quantization_observer_epoch: 5.0
    end_epoch: -1.0
    freeze_bn_stats_epoch: 5.0
    quantize_embeddings: 1
    start_epoch: 0.0
    submodules: ['bert.encoder', 'qa_outputs', 'bert.embeddings']