| MODEL_INFO = { |
| "name": "Neural Cellular Automata", |
| "short_name": "NCA", |
| "category": "Cellular Automata", |
| "description": "Learns local cellular update rules that grow images from a single living seed.", |
| "supports_training": True, |
| "supports_generation": True, |
| "order": 10, |
| } |
|
|
| TRAINING_SETTINGS = { |
| "training_mode": { |
| "label": "Training mode", |
| "type": "choice", |
| "options": ["Single target", "Dataset average (experimental)"], |
| "default": "Single target", |
| }, |
| "target_image": {"label": "Target image", "type": "file", "default": ""}, |
| "epochs": {"label": "Training steps", "type": "int", "default": 500, "min": 1, "max": 100000}, |
| "batch_size": {"type": "int", "default": 8, "min": 1, "max": 128}, |
| "learning_rate": {"type": "float", "default": 0.001, "min": 0.0000001}, |
| "resolution": {"label": "Target resolution", "type": "choice", "options": [32, 64, 128], "default": 64}, |
| "cell_channels": {"type": "int", "default": 16, "min": 4, "max": 64}, |
| "hidden_size": {"type": "int", "default": 128, "min": 16, "max": 512}, |
| "min_growth_steps": {"type": "int", "default": 64, "min": 1, "max": 512}, |
| "max_growth_steps": {"type": "int", "default": 96, "min": 1, "max": 1024}, |
| "stable_growth": {"label": "Stable growth training", "type": "bool", "default": False}, |
| "stable_max_growth_steps": {"label": "Stable maximum growth steps", "type": "int", "default": 160, "min": 1, "max": 2048}, |
| "fire_rate": {"label": "Update probability", "type": "slider", "default": 0.5, "min": 0.05, "max": 1.0, "resolution": 0.05}, |
| "random_seed": {"type": "int", "default": 42}, |
| "mixed_precision": {"type": "bool", "default": True}, |
| "device": {"type": "choice", "options": ["auto", "cuda", "cpu"], "default": "auto"}, |
| "preview_every": {"label": "Preview every N steps", "type": "int", "default": 10, "min": 1}, |
| "checkpoint_every": {"label": "Checkpoint every N steps", "type": "int", "default": 50, "min": 1}, |
| "loss_smoothing_window": {"label": "Loss smoothing window", "type": "int", "default": 25, "min": 1, "max": 1000}, |
| "fit_mode": {"label": "Image fit", "type": "choice", "options": ["contain", "crop", "stretch"], "default": "contain"}, |
| "background": {"type": "choice", "options": ["transparent", "white", "black"], "default": "transparent"}, |
| "horizontal_flip": {"label": "Random horizontal flip", "type": "bool", "default": False}, |
| "resume_checkpoint": {"label": "Resume checkpoint (optional)", "type": "file", "default": ""}, |
| } |
|
|
| GENERATION_SETTINGS = { |
| "growth_steps": {"type": "int", "default": 100, "min": 1, "max": 5000}, |
| "seed": {"type": "int", "default": -1}, |
| "fire_rate": {"label": "Update probability", "type": "slider", "default": 0.5, "min": 0.05, "max": 1.0, "resolution": 0.05}, |
| "animate_growth": {"type": "bool", "default": True}, |
| "frame_every": {"label": "Save every Nth step", "type": "int", "default": 5, "min": 1}, |
| "export_gif": {"type": "bool", "default": True}, |
| } |
|
|
