| { | |
| "architectures": [ | |
| "CodelorColor" | |
| ], | |
| "model_type": "codelor-color", | |
| "format": "codelor_color_standalone_v2", | |
| "version": "1.2", | |
| "task": "image-segmentation", | |
| "labels": [ | |
| "background", | |
| "green" | |
| ], | |
| "green_ref": [ | |
| 0.2, | |
| 0.85, | |
| 0.3 | |
| ], | |
| "hyperparameters": { | |
| "dim": 128, | |
| "depth": 8, | |
| "heads": 8, | |
| "tied_after": 4, | |
| "mlp_ratio": 3.0, | |
| "max_tokens": 1024 | |
| }, | |
| "training": { | |
| "best_loss": 0.17389452453454335, | |
| "epochs_done": 30, | |
| "params": 1641055 | |
| } | |
| } |