train_config.json

{
  "CUDA_VISIBLE_DEVICES": "1",
  "max_RAM_usage": 85,

  "data": {
    "train_dirs": ["./DCASE_2023_Challenge_Task_7_Dataset/train.txt"],
    "train_cond_dirs": null,
    "test_dirs": ["./DCASE_2023_Challenge_Task_7_Dataset/eval.txt"],
    "test_cond_dirs": null,
    "sample_rate": 22050,
    "audio_length": 88200,
    "n_workers": 4
  },

  "model": {
    "model_dir": "train_logs",
    "sequential": "mamba",
    "factors": [2, 2, 3, 5, 5],
    "dims":  [16, 32, 64, 128, 256, 512],
    "bottleneck_layers": 1,
    "bidirectional_bottleneck": true
  },

  "condition": {
    "time_emb_dim": 512,
    "class_emb_dim": 512,
    "mid_dim": 512,
    "film_type": "mamba",
    "block_nums": [49, 49, 49, 49, 49, 49, 14],
    "event_type": "rms",
    "event_dims": {
      "rms": 690,
      "power": 88200,
      "onset": 88200
    },
    "cond_prob": [0.1, 0.1]
  },

  "training": {
    "lr": 0.0001,
    "batch_size": 2,
    "ema_rate": 0.999,
    "scheduler_patience_epoch": 25,
    "scheduler_factor": 0.8,
    "scheduler_threshold": 0.001,
    "n_epochs": 500
  },

  "logging": {
    "checkpoint_id": null,
    "n_epochs_to_checkpoint": 15,
    "n_epochs_to_log": 15,
    "n_steps_to_log": 300,
    "n_steps_to_test": 300,
    "n_bins": 5
  },

  "profiler": {
    "use_profiler": false,
    "wait": 298,
    "warmup": 2,
    "active": 3,
    "repeat": 0
  },

  "telegram": {
    "apiToken": "",
    "chatID": ""
  }
}