Method load_pretrained_models

ram/models/ram_base_model.py:40–44

(self)

        self.setup_loss_functions()
        
    def load_pretrained_models(self):
        load_path = self.opt['path'].get('pretrain_network_g', None)
        if load_path is not None:
            param_key = self.opt['path'].get('param_key_g', 'params')
            self.load_network(self.net_g, load_path, self.opt['path'].get('strict_load_g', True), param_key)

    def init_ema_model(self):
        logger = get_root_logger()