diff --git a/docs/apis/load_model.md b/docs/apis/load_model.md index 9a32ca613cb61c72aea0d1822c7fba6f26846472..0e460ee583c8bb4030d202372dae40882b7b34dd 100644 --- a/docs/apis/load_model.md +++ b/docs/apis/load_model.md @@ -34,7 +34,7 @@ pred_result = model.predict('./xiaoduxiong_ins_det/JPEGImages/WechatIMG114.jpeg' # 在验证集上进行评估 eval_reader = pdx.cv.datasets.CocoDetection(data_dir=data_dir, - ann_file=ann_file + ann_file=ann_file, transforms=model.eval_transforms) eval_result = model.evaluate(eval_reader, batch_size=1) ```