No padding in colloator

2023-04-02 02:05:03 +08:00 · 2023-04-02 02:05:03 +08:00 · 4227999d4c
parent c508f62b70
commit 4227999d4c
1 changed files with 1 additions and 0 deletions
--- a/ptuning/main.py
+++ b/ptuning/main.py
@ -262,6 +262,7 @@ def main():
        model=model,
        label_pad_token_id=label_pad_token_id,
        pad_to_multiple_of=None,
+        padding=False
    )

    # Metric