config.json

{
  "model_name": "facebook/bart-base",
  "k": 384,
  "pool_layers": 2,
  "pool_heads": null,
  "pool_type": "conv",
  "d_ae": 768,
  "recon_layers": 2,
  "recon_heads": null,
  "train_decoder": true
}