[BERT] Pre-trained checkpoint conversion script across frameworks

mlcommons / training_policies

Issues related to MLPerf™ training policies, including rules and suggested changes

https://mlcommons.org/en/groups/training

Apache License 2.0

92 stars 66 forks source link

[BERT] Pre-trained checkpoint conversion script across frameworks #419

Open nv-rborkar opened 3 years ago

nv-rborkar commented 3 years ago

As per #317 , we would like all references that use pre-trained weights or start from a checkpoint to provide conversion scripts for use across different frameworks (Pytorch, TF, & other frameworks of choice for submission) to ensure submitters use the same pre-approved checkpoint.

AI Google (reference owner) for BERT Example script & recipe in #391

johntran-nv commented 3 years ago

@petermattson could you point this to the correct google person?