File size: 484 Bytes
ab82f5a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
!LayerPruningModifier
end_epoch: -1.0
layers: ['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.embeddings', 'qa_outputs', 'bert.encoder']
|