config.json

{
  "_attn_implementation_autoset": true,
  "algorithm_name": "Mixup",
  "dataset_name": "ColoredMNIST",
  "hparams": {
    "batch_size": 64,
    "class_balanced": false,
    "data_augmentation": true,
    "dinov2": false,
    "freeze_bn": false,
    "lars": false,
    "linear_steps": 500,
    "lr": 0.001,
    "mixup_alpha": 0.2,
    "nonlinear_classifier": false,
    "resnet18": false,
    "resnet50_augmix": true,
    "resnet_dropout": 0.0,
    "vit": false,
    "vit_attn_tune": false,
    "vit_dropout": 0.0,
    "weight_decay": 0.0
  },
  "input_shape": [
    2,
    28,
    28
  ],
  "model_type": "domainbed",
  "num_classes": 2,
  "num_domains": 1,
  "test_envs": [
    0,
    1
  ],
  "transformers_version": "4.46.2"
}