|
|
|
@@ -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") |
|
|
|
|