add model zoo

[ChenMinQi]

🚀 Feature

model zoo just like detectron2

Motivation

We can quickly know the accuracy/train time/inference time of each model.

Pitch

just like this, detectron2 model zoo

Alternatives

N/A

Additional context

N/A

[dsp6414]

i use torchvision.models for model zoo [code] def Create_Model_From_Zoo(num_classes=4): model = models.resnet50(pretrained=True) num_ftrs = model.fc.in_features model.fc = nn.Linear(num_ftrs,num_classes) classy_model = ClassyModel.from_model(model) return classy_model

... config_file='./configs/test.json' config = load_json(config_file)

model = Create_Model_From_Zoo(num_classes=4) task = ( My_ClassificationTask() .set_loss(build_loss(config["loss"])) .set_model(model) .set_optimizer(build_optimizer(config["optimizer"])) .set_meters([AccuracyMeter(topk=[1])]) )

...