-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[v1.x] Add AWDRNN Pratrained model test #20018
Conversation
Hey @Zha0q1 , Thanks for submitting the PR
CI supported jobs: [edge, centos-gpu, centos-cpu, windows-cpu, sanity, miscellaneous, website, windows-gpu, unix-cpu, clang, unix-gpu] Note: |
|
||
@with_seed() | ||
@pytest.mark.parametrize('model_name', [('awd_lstm_lm_600', 600), ('awd_lstm_lm_1150', 1150)]) | ||
@pytest.mark.parametrize('seq_length', [16, 32]) |
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.
Shall we also test other seq_length like 128 or 256?
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.
Yeah I can change them to 96/128, which are two commonly used lengths
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
No description provided.