Views
No views yet
generators.pt — state dicts for both generators (G_ab, G_ba) plus training args.1import torch
2from huggingface_hub import hf_hub_download
3from cyclegan import Generator
4
5ckpt_path = hf_hub_download("mmkuznecov/cyclegan-summer2winter", "generators.pt")
6ckpt = torch.load(ckpt_path, map_location="cpu")
7
8G_ab = Generator(n_residual=ckpt["args"]["n_residual"])
9G_ab.load_state_dict(ckpt["G_ab"])
10G_ab.eval()data_dir: ./data/summer2winter_yosemite
out_dir: ./checkpoints
epochs: 20
image_size: 128
batch_size: 4
lr: 0.0002
beta1: 0.5
lambda_cycle: 10.0
lambda_identity: 5.0
n_residual: 6
num_workers: 2
sample_every: 1
seed: 42