{
  "common": {
    "dataset_path": "raw_data",
    "processed_dataset_path": "processed_data",
    "runs_dir": "runs",
    "datasets": [
      "birds",
      "creatures"
    ],
    "clip_model_path": "processed_data/hf/openai_clip-vit-base-patch32",
    "device": "auto",
    "seed": 4267114,
    "use_tensorboard": true,
    "run_name": "new0002",
    "eval_ratio": 0.1,
    "test_ratio": 0.0,
    "eval_every": 5,
    "reconstruction_samples": 4
  },
  "preprocess": {
    "keypoint_epsilon": 0.015,
    "max_raw_stroke_points": 1500,
    "max_keypoints": 48,
    "min_keypoints": 2
  },
  "generator": {
    "epochs": 100,
    "batch_size": 256,
    "num_workers": 16,
    "save_every": 5,
    "lr": 0.0003,
    "warmup_epochs": 5,
    "min_lr_factor": 0.01,
    "weight_decay": 0.01,
    "amp": true,
    "amp_dtype": "bf16",
    "allow_tf32": true,
    "non_blocking_transfer": true,
    "max_outer_tokens": 1024,
    "grid_size": 256,
    "num_delta_bins": 129,
    "delta_quantile": 0.99,
    "coordinate_soft_label_sigma": 1.0,
    "coordinate_soft_label_radius": 4,
    "text_encoder_model": "openai/clip-vit-base-patch32",
    "text_embedding_dim": 512,
    "text_embedding_batch_size": 256,
    "tag_clip_embedding_dim": 512,
    "prompt_text_stroke_prob": 0.4,
    "prompt_text_only_prob": 0.25,
    "prompt_stroke_only_prob": 0.25,
    "prompt_empty_prob": 0.1,
    "d_model": 256,
    "nhead": 8,
    "num_layers": 8,
    "dropout": 0.1
  }
}