hcn/config.json

{
  "num_age_bins": 5,
  "num_sex": 2,
  "num_race": 3,
  "d_node": 256,
  "d_ctx": 1024,
  "dropout": 0.1,
  "use_uncertainty": true,
  "use_aux_loss": true,
  "aux_hidden_dim": 512,
  "encode_age": false
}