Is there a plan to add a registration mechanism to torchvision.models?

[zhiqwang]

Hi,
As mentioned in the test scripts:

vision/test/test_models.py

Line 9 in c187c2b

# TODO add a registration mechanism to torchvision.models

I want to know whether the registration mechanism is adding to ./torchvision/models/resnet.py or somewhere like this?

[fmassa]

This is not yet 100% clear to me that it should be done.
But by registration mechanism I meant adding a way of retrieving all possible models from a particular category (say detection, or video). We currently have some hacks that yields everything under torchvision.models that are functions and do not start with an underscore, but this is fragile and might be better to consider a more robust way of doing this.