experiment results loader

[Ootzk]

at analyze.py

[Ootzk]

def load_pretrained_model(ID: int, checkpoint: int, verbose: bool):
    experiment_dir = f'experiments/{ID}'
    with open(f'{experiment_dir}/config.json', 'r') as exp_config_file:
        exp_config = json.load(exp_config_file)
        
    if verbose:
        print(f'experiment configuration: ')
        pprint.pprint(exp_config)
        
    model = load_model(exp_config)
    for candidate in os.listdir(f'{experiment_dir}/checkpoints/'):
        if candidate.startswith(f'checkpoint_{checkpoint}_'):
            model.load_state_dict(torch.load(f'{experiment_dir}/checkpoints/{candidate}')['model'])
            if verbose:
                print(f'checkpoint: epoch @ {checkpoint} loaded successfully! return pretrained model and configuration.')
            return model, exp_config
        
    if verbose:
        print(f'checkpoint: epoch @ {checkpoint} not searched; just return skeleton and configuration.')
        return model, exp_config

reference model loader