Update comfunc.py

2021-05-31 14:28:22 +08:00 · 2021-05-31 14:28:22 +08:00 · b79f3f2e65
parent e9a1d83035
commit b79f3f2e65
1 changed files with 1 additions and 1 deletions
--- a/ppcls/losses/comfunc.py
+++ b/ppcls/losses/comfunc.py
@ -41,4 +41,4 @@ def rerange_index(batch_size, samples_each_class):
        rerange_index += (pos_idx + neg_idx)

    rerange_index = np.array(rerange_index).astype(np.int32)
-    return rerange_index
+    return rerange_index