Exemplos de create_base_model em Python

Exemplo n.º 1

Arquivo: basetrainer.py Projeto: uds-lsv/transfer-distant-transformer-african

    def __init__(self, exp_config):
        model = create_base_model(exp_config)
        super().__init__(model, exp_config)

        num_parameters = np.sum(
            [p.numel() for p in self.main_model.parameters()])
        logger.info(f"The model has {num_parameters} parameters.")
        raise Exception("Done")

Exemplo n.º 2

Arquivo: basetrainer.py Projeto: uds-lsv/transfer-distant-transformer-african

 def __init__(self, exp_config):
     model = create_base_model(exp_config)
     super().__init__(model, exp_config)

Exemplo n.º 3

Arquivo: evaltrainer.py Projeto: uds-lsv/transfer-distant-transformer-african

 def __init__(self, exp_config):
     self._load_config(exp_config)
     model = create_base_model(exp_config)
     super().__init__(model, exp_config)