-
Notifications
You must be signed in to change notification settings - Fork 86
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
Allow model_fn to get more arguments #65
Comments
Any update on this issue? This is limiting as when a user wants to pass different hyperparameters, they must be hardcoded inside the |
I agree that the single |
Right now,
model_fn
only acceptsmodel_dir
which is very limiting in case a model needs some arguments to be instantiated first then can load the checkpoint.The text was updated successfully, but these errors were encountered: