def get_network()

in src/builder.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

    def get_network(self):
        # If network already loaded simply return
        if self._network is not None: return self._network

        self._logger.info("Retrieving model")

        # If checkpoint file is available, load from checkpoint
        state_dict = self.get_trainer().try_load_statedict_from_checkpoint()

        self._network = BertModel(self._bert_model_name, self.get_label_mapper().num_classes,
                                  fine_tune=self.fine_tune, bert_config=self._bert_config)

        if state_dict is not None:
            # Only load from BERT pretrained when no checkpoint is available
            self._logger.info("checkpoint models found")
            self._network.load_state_dict(state_dict)

        self._logger.info("Retrieving model complete")


        return self._network