You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

org_resnet.py 439B

1234567891011121314
  1. from typing import Tuple
  2. from ...configs.rsna_configs import RSNAConfigs
  3. from ...models.model import Model
  4. from ..entrypoint import BaseEntrypoint
  5. from ...models.rsna.org_resnet import RSNAORGResNet18
  6. class EntryPoint(BaseEntrypoint):
  7. def _get_conf_model(self) -> Tuple[RSNAConfigs, Model]:
  8. config = RSNAConfigs('RSNA_ResNet_Org', 1, 224, self.phase_type)
  9. model = RSNAORGResNet18()
  10. return config, model