diff --git a/official/cv/resnet/golden_stick/quantization/slb/resnet18_cifar10_config.yaml b/official/cv/resnet/golden_stick/quantization/slb/resnet18_cifar10_config.yaml index 4ff87a3c837836849c83279ed54c495875006ea5..613ddaa4a0473d4a5285b3b6b58bc872f4df8aaf 100644 --- a/official/cv/resnet/golden_stick/quantization/slb/resnet18_cifar10_config.yaml +++ b/official/cv/resnet/golden_stick/quantization/slb/resnet18_cifar10_config.yaml @@ -102,3 +102,9 @@ checkpoint_path: "The location of the checkpoint file." checkpoint_file_path: "The location of the checkpoint file." save_graphs: "Whether save graphs during training, default: False." save_graphs_path: "Path to save graphs." +comp_algo: "The quantization algorithm, default: SLB" +quant_type: "Quantization bit number of weight, available: [W1, W2, W4]" +t_start_val: "Initial value of temperature hyperparameters, default: 1.0" +t_start_time: "Fraction of epochs after which temperature hyperparameters starting changing, default: 0.2" +t_end_time: "Fraction of epochs after which temperature hyperparameters stopping changing, default: 0.6" +t_factor: "Multiplicative factor of temperature hyperparameters changing, default: 1.2"