|
|
|
@ -50,7 +50,8 @@ if __name__ == "__main__":
|
|
|
|
|
# define fusion network
|
|
|
|
|
network = LeNet5Fusion(cfg.num_classes)
|
|
|
|
|
# convert fusion network to quantization aware network
|
|
|
|
|
network = quant.convert_quant_network(network, quant_delay=0, bn_fold=False, freeze_bn=10000)
|
|
|
|
|
network = quant.convert_quant_network(network, quant_delay=0, bn_fold=False, freeze_bn=10000,
|
|
|
|
|
per_channel=[True, False])
|
|
|
|
|
|
|
|
|
|
# define loss
|
|
|
|
|
net_loss = nn.SoftmaxCrossEntropyWithLogits(is_grad=False, sparse=True, reduction="mean")
|
|
|
|
|