def build_trainer(configuration, *rest, **kwargs): configuration.freeze() config = configuration.get_config() registry.register("config", config) registry.register("configuration", configuration) trainer_type = config.training.trainer trainer_cls = registry.get_trainer_class(trainer_type) trainer_obj = trainer_cls(configuration) # Set args as an attribute for future use trainer_obj.args = configuration.args return trainer_obj
def build_trainer(config: DictConfig) -> Any: """Builder function for creating a trainer class. Trainer class name is picked from the config. Args: config (DictConfig): Configuration that will be used to create the trainer. Returns: (BaseTrainer): A trainer instance """ trainer_type = config.training.trainer trainer_cls = registry.get_trainer_class(trainer_type) trainer_obj = trainer_cls(config) return trainer_obj