Skip to content

Commit 65b0e54

Browse files
committed
docs: 新增注释
1 parent bf687c9 commit 65b0e54

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

code/chapter-8/01_classification/resnet_ptq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ def ptq(args):
192192
classes = ["NORMAL", "PNEUMONIA"]
193193
criterion = nn.CrossEntropyLoss() # 选择损失函数
194194
loss_m_valid, acc_m_valid, mat_valid = utils.ModelTrainer.evaluate(valid_loader, model, criterion, device, classes)
195-
logger.info('PTQ量化后模型ACC :{}'.format(acc_m_valid.avg))
195+
logger.info('PTQ量化后模型ACC :{},scale值计算方法是:{}'.format(acc_m_valid.avg, args.ptq_method))
196196
# ------------------------------------ step5: 保存ptq量化后模型 ------------------------------------
197197
dir_name = os.path.dirname(args.ckpt_path)
198198
ptq_ckpt_path = os.path.join(dir_name, "resnet50_ptq.pth")

0 commit comments

Comments
 (0)