| { | |
| "model_type": "film_unet2d", | |
| "architectures": ["FilmUnet2DModel"], | |
| "auto_map": { | |
| "AutoConfig": "configuration_film_unet2d.FilmUnet2DConfig", | |
| "AutoModel": "modeling_film_unet2d.FilmUnet2DModel", | |
| "AutoImageProcessor": "image_processing_film_unet2d.FilmUnet2DImageProcessor" | |
| }, | |
| "in_channels": 3, | |
| "num_classes": 1, | |
| "n_organs": 9, | |
| "size": 32, | |
| "depth": 5, | |
| "film_start": 0, | |
| "use_film": 1 | |
| } | |