| { |
| "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 |
| } |
| } |
