train_swin_transformer.json

{
  "name": "swin_transformer",
  "n_gpu": 1,
  "arch": {
    "type": "swin_transformer",
    "args": {
      "depths": [
        2,
        2,
        6,
        2
      ],
      "num_heads": [
        8,
        16,
        16,
        24
      ],
      "in_chans": 8,
      "embed_dim": 96,
      "patch_size":4,
      "window_size": [7,7,7,7],
      "img_size": 3584,
       "num_classes": 108
    }
  },

  "data_loader": {
    "type": "ChallengeDataLoader",
      "args": {
        "label_dir": "/data/ecg/raw_data/challenge2020/bak_all_data_2020",
        "split_index": "process/data_split/split1.mat",
        "save_dir": "/data/ecg/challenge2020/data",
        "lead_number": 8,
        "batch_size": 100,
        "resample_Fs": 300,
        "window_size":3584,
        "normalization": false,
        "augmentations": null,
        "save_data": true,
        "load_saved_data": false,
        "p": 0.8
    }
  },

  "optimizer": {
    "type": "AdamW",
    "args": {
      "lr": 0.0001
    }
  },
  "loss": {
    "type": "bce_with_logits_loss"
  },
  "metrics": [
    "accuracy",
    "f_measure",
    "macro_f_beta_measure",
    "macro_g_beta_measure",
    "macro_auroc",
    "macro_auprc",
    "challenge_metric"
  ],
  "lr_scheduler": {
    "type": "StepLR",
    "args": {
      "step_size": 20,
      "gamma": 0.1
    }
  },
 "trainer": {
        "epochs": 1000,
        "save_dir": "/data/ecg/challenge2020/saved",
        "save_period": 1,
        "verbosity": 2,
        "monitor": "max val_challenge_metric",
        "early_stop": 5,
        "tensorboard": false,
        "only_scored_class": true
    },
  "evaluater": {
      "result_dir": "/data/ecg/challenge2020/saved",
      "verbosity": 2,
      "visualization": {
          "scatter": true
      },
     "n_segment": 1
  }

}