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