train_config.json

{
  "N_max": 100,
  "batch_size": 16,
  "beta_1": 0.9,
  "beta_2": 0.999,
  "data_aug_h_flip": 1,
  "data_aug_rot": 1,
  "data_aug_v_flip": 1,
  "dropout": 0.46575581405192834,
  "epochs": 30,
  "fc1_neurons": 512,
  "fc2_neurons": 32,
  "generate_embeddings": 0,
  "learning_rate": 5e-07,
  "loss": "binary_crossentropy",
  "model_kind": "hf_hub:mwalmsley/baseline-encoder-regression-maxvit_tiny",
  "model_name": "MaxViT",
  "patience": 4,
  "pretrained": 1,
  "random_seed": 0,
  "testing": 0,
  "train_data_version": "v11",
  "warmup_epochs": 6
}