config.json

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