(self, device, distributed=False, local_rank: int = 0)
| 27 | return self.model + '-' + getattr(structure_model, self.model).model_name(self.model_args) |
| 28 | |
| 29 | def build(self, device, distributed=False, local_rank: int = 0): |
| 30 | Model = getattr(structure_model, self.model) |
| 31 | model = Model(self.model_args, device, |
| 32 | distributed=distributed, local_rank=local_rank) |
| 33 | return model |
| 34 | |
| 35 | class Detector(Configurable): |
| 36 | structure = State(autoload=False) |
no test coverage detected