|
import pickle |
|
import functools |
|
import torch |
|
from configs import paths_config, global_config |
|
|
|
|
|
def toogle_grad(model, flag=True): |
|
for p in model.parameters(): |
|
p.requires_grad = flag |
|
|
|
|
|
def load_tuned_G(run_id, type): |
|
new_G_path = f'{paths_config.checkpoints_dir}/model_{run_id}_{type}.pt' |
|
with open(new_G_path, 'rb') as f: |
|
new_G = torch.load(f).to(global_config.device).eval() |
|
new_G = new_G.float() |
|
toogle_grad(new_G, False) |
|
return new_G |
|
|
|
|
|
def load_old_G(): |
|
with open(paths_config.stylegan2_ada_ffhq, 'rb') as f: |
|
old_G = pickle.load(f)['G_ema'].to(global_config.device).eval() |
|
old_G = old_G.float() |
|
return old_G |
|
|