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']
|