Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
e150c4e2
"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "e5f3487190869e043de00ce1a51f47c5b79009e2"
Unverified
Commit
e150c4e2
authored
Oct 10, 2022
by
Kaiyu Yang
Committed by
GitHub
Oct 10, 2022
Browse files
Fix the error message in run_t5_mlm_flax.py (#19282)
parent
e3f028f3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
examples/flax/language-modeling/run_t5_mlm_flax.py
examples/flax/language-modeling/run_t5_mlm_flax.py
+1
-1
No files found.
examples/flax/language-modeling/run_t5_mlm_flax.py
View file @
e150c4e2
...
...
@@ -349,7 +349,7 @@ class FlaxDataCollatorForT5MLM:
if
batch
[
"input_ids"
].
shape
[
-
1
]
!=
self
.
input_length
:
raise
ValueError
(
f
"`input_ids` are incorrectly preprocessed. `input_ids` length is
{
batch
[
'input_ids'
].
shape
[
-
1
]
}
, but"
f
" should be
{
self
.
targe
t_length
}
."
f
" should be
{
self
.
inpu
t_length
}
."
)
if
batch
[
"labels"
].
shape
[
-
1
]
!=
self
.
target_length
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment