Update references to use the new Model Registration API #6369
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Depends on #6364
This PR updates all the
__dict__[model_name]
occurrences within our code with aget_model()
call.To confirm the reference scripts still work we run the same experiments as #5618 and #6241. Minor variations on accuracy are expected because we use multiple GPUs and batch-sizes > 1. The important think to verify is that the models are initialized properly and provide similar accuracies.