|
|
@ -170,7 +170,8 @@ def train():
|
|
|
|
if config.quantization_aware:
|
|
|
|
if config.quantization_aware:
|
|
|
|
quantizer = QuantizationAwareTraining(bn_fold=True,
|
|
|
|
quantizer = QuantizationAwareTraining(bn_fold=True,
|
|
|
|
per_channel=[True, False],
|
|
|
|
per_channel=[True, False],
|
|
|
|
symmetric=[True, False])
|
|
|
|
symmetric=[True, False],
|
|
|
|
|
|
|
|
one_conv_fold=False)
|
|
|
|
network = quantizer.quantize(network)
|
|
|
|
network = quantizer.quantize(network)
|
|
|
|
|
|
|
|
|
|
|
|
network = YoloWithLossCell(network)
|
|
|
|
network = YoloWithLossCell(network)
|
|
|
|