LAP/torchlap/experiments/rsna/org_inception.py

from typing import Tuple
from ...configs.rsna_configs import RSNAConfigs
from ...models.model import Model
from ..entrypoint import BaseEntrypoint
from ...models.rsna.org_inception import RSNAORGInception


class EntryPoint(BaseEntrypoint):
    
    def _get_conf_model(self) -> Tuple[RSNAConfigs, Model]:
        config = RSNAConfigs('RSNA_Inception_Org', 2, 299, self.phase_type)
        model = RSNAORGInception(0.4)
        return config, model