config.json

{
  "device": "cpu",
  "epsilon_scalar_enc": 1.1,
  "hidden_dim": 256,
  "hidden_dim_scalar_enc": 32,
  "kernel_size": 41,
  "num_patches": 32,
  "output_token": "cls_token",
  "pre_training": false,
  "return_transf_layer": -1,
  "scalar_scales": null,
  "transf_depth": 6,
  "transf_dim_head": 32,
  "transf_dropout": 0.1,
  "transf_mlp_dim": 512,
  "transf_num_heads": 8
}