@@ -32,7 +32,7 @@ eval_dataset = pdx.datasets.ImageNet(
# 加载模型
model = pdx.load_model('output/mobilenet_v2/best_model')
-# 量化训练
+# 在线量化
model.quant_aware_train(
num_epochs=5,
train_dataset=train_dataset,
@@ -41,7 +41,7 @@ eval_dataset = pdx.datasets.VOCDetection(
model = pdx.load_model('output/yolov3_darknet53/best_model')
num_epochs=50,
@@ -39,7 +39,7 @@ eval_dataset = pdx.datasets.SegDataset(
model = pdx.load_model('output/unet/best_model')