def METHOD_NAME(self):
    if self.args.backbone_freezed:
        logging.info("Initializing model; Backbone Freezed")
        return self.model.encoder_decoder.cpu().state_dict()
    else:
        logging.info("Initializing end-to-end model")
        return self.model.cpu().state_dict()