Skip to content
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

[Minigo] Pre-trained checkpoint conversion script across frameworks #420

Open
nv-rborkar opened this issue Jan 20, 2021 · 4 comments
Open
Assignees
Labels
AI There is an action item here. Next Meeting Item to be discussed in the next Working Group

Comments

@nv-rborkar
Copy link
Contributor

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 Intel (reference owner) for Minigo
Example script & recipe in #391

@johntran-nv johntran-nv added AI There is an action item here. Next Meeting Item to be discussed in the next Working Group labels Jan 20, 2021
@johntran-nv
Copy link
Contributor

@christ1ne could you assign to the reference owner?

@christ1ne
Copy link

@johntran-nv are you referring to the following example for converting the weights? mlcommons/training#436
That's a from PyTorch model. Is there an TF model weights export example?
Thanks!

@delock who's the reference owner.

@christ1ne christ1ne self-assigned this Jan 20, 2021
@johntran-nv
Copy link
Contributor

Yes that's what we're referring to. Sorry but I don't know of a similar TF export example. We could ask the Google folks for help, if you want.

@christ1ne
Copy link

I think we should ask the Google folks for an TF example.
Meanwhile, I am not sure if all the TF models can be converted to Pytorch as they are today.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
AI There is an action item here. Next Meeting Item to be discussed in the next Working Group
Projects
None yet
Development

No branches or pull requests

3 participants