-
Notifications
You must be signed in to change notification settings - Fork 27.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Skip equivalence test for TransfoXL #16224
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
tests/test_modeling_tf_common.py
Outdated
@@ -476,7 +476,7 @@ def check_pt_tf_models(tf_model, pt_model): | |||
"TFFunnelForPreTraining", | |||
"TFElectraForPreTraining", | |||
"TFXLMWithLMHeadModel", | |||
]: | |||
] + ["TFTransfoXLLMHeadModel"]: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Doesn't it make more sense to add it in the list?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Copied it from above, but you're right! Will fix both.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done in bb5874a
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I separate it to highlight the issue for TFTransfoXLLMHeadModel
is somehow different than in other models. (losses
). But I am OK for this change.
The documentation is not available anymore as the PR was closed or merged. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
* Skip test for TransfoXL * Single list
* Skip test for TransfoXL * Single list
Skips the equivalence test for TransfoXL as it does not have the
loss
orlosses
returned.