stage2_config.json

{
  "model_checkpoint_path": "None",
  "model_type": "pfam",
  "fast_dev_run": 0,
  "loss_type": "MMD",
  "dataset_type": "default",
  "precision": "32",
  "stage1_dataset_path": "None",
  "stage2_output_path": "None",
  "seed": 42,
  "num_workers": 12,
  "dropout": 0.0,
  "batch_size": 64,
  "emb_dim": 512,
  "hid_dim": 1024
}