(self)
| 67 | print(MODELS) |
| 68 | |
| 69 | def test_build(self): |
| 70 | MODELS = Registry('models') |
| 71 | |
| 72 | @MODELS.register_module(Tasks.image_classification, 'SwinT') |
| 73 | class SwinTForCls(object): |
| 74 | pass |
| 75 | |
| 76 | @MODELS.register_module(Tasks.sentiment_analysis, 'Bert') |
| 77 | class BertForSentimentAnalysis(object): |
| 78 | pass |
| 79 | |
| 80 | cfg = dict(type='SwinT') |
| 81 | model = build_from_cfg(cfg, MODELS, Tasks.image_classification) |
| 82 | self.assertTrue(isinstance(model, SwinTForCls)) |
| 83 | |
| 84 | cfg = dict(type='Bert') |
| 85 | model = build_from_cfg(cfg, MODELS, Tasks.sentiment_analysis) |
| 86 | self.assertTrue(isinstance(model, BertForSentimentAnalysis)) |
| 87 | |
| 88 | with self.assertRaises(KeyError): |
| 89 | cfg = dict(type='Bert') |
| 90 | model = build_from_cfg(cfg, MODELS, Tasks.image_classification) |
| 91 | |
| 92 | |
| 93 | if __name__ == '__main__': |
nothing calls this directly
no test coverage detected