Merge pull request #1448 from TingquanGao/2.3/fix_link

[cherry-pick] 2.3/fix link

Merge pull request #1448 from TingquanGao/2.3/fix_link
[cherry-pick] 2.3/fix link
74e934a6 · cuicheng01 · GitHub · 66227977 · 968ff751 · 74e934a6
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

ppcls/arch/backbone/model_zoo/rexnet.py ppcls/arch/backbone/model_zoo/rexnet.py +1 -1

ppcls/engine/evaluation/classification.py ppcls/engine/evaluation/classification.py +1 -1

未找到文件。
--- a/ppcls/arch/backbone/model_zoo/rexnet.py
+++ b/ppcls/arch/backbone/model_zoo/rexnet.py
@@ -30,7 +30,7 @@ MODEL_URLS = {
    "ReXNet_1_3":
    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ReXNet_1_3_pretrained.pdparams",
    "ReXNet_1_5":
-    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ReXNet_1_5_32x4d_pretrained.pdparams",
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ReXNet_1_5_pretrained.pdparams",
    "ReXNet_2_0":
    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ReXNet_2_0_pretrained.pdparams",
    "ReXNet_3_0":

--- a/ppcls/engine/evaluation/classification.py
+++ b/ppcls/engine/evaluation/classification.py
@@ -91,7 +91,7 @@ def classification_eval(engine, epoch_id=0):
                    paddle.distributed.all_gather(pred_list, out)
                    pred = paddle.concat(pred_list, 0)
-                if accum_samples > total_samples:
+                if accum_samples > total_samples and not engine.use_dali:
                    pred = pred[:total_samples + current_samples -
                                accum_samples]
                    labels = labels[:total_samples + current_samples -