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.4595202732711804,
  "epochs": 30,
  "fc1_neurons": 128,
  "fc2_neurons": 16,
  "generate_embeddings": 0,
  "learning_rate": 5e-06,
  "loss": "binary_crossentropy",
  "model_kind": "maxvit_tiny_rw_224.sw_in1k",
  "model_name": "MaxViT",
  "patience": 4,
  "pretrained": 0,
  "random_seed": 5,
  "testing": 0,
  "train_data_version": "v11",
  "warmup_epochs": 4
}