CycleGAN — cyclegan-summer2winter

CycleGAN generators trained with the code in this repo.

Files

• generators.pt — state dicts for both generators (G_ab, G_ba) plus training args.

Usage

import torch
from huggingface_hub import hf_hub_download
from cyclegan import Generator

ckpt_path = hf_hub_download("mmkuznecov/cyclegan-summer2winter", "generators.pt")
ckpt = torch.load(ckpt_path, map_location="cpu")

G_ab = Generator(n_residual=ckpt["args"]["n_residual"])
G_ab.load_state_dict(ckpt["G_ab"])
G_ab.eval()

Training args

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