domainbed-4bfa7f7c027a74878ce6cd0f6b09b0b4

DomainBed Model

This model was trained using the DomainBed framework.

Model Information

• Model ID: domainbed-4bfa7f7c027a74878ce6cd0f6b09b0b4
• Model Path: /my/sweep/output/path/4bfa7f7c027a74878ce6cd0f6b09b0b4
• Model File: model.pkl

Training Results

{
  "args": {
    "algorithm": "IRM",
    "checkpoint_freq": null,
    "data_dir": "./domainbed/data/",
    "dataset": "PACS",
    "holdout_fraction": 0.2,
    "hparams": null,
    "hparams_seed": 1,
    "output_dir": "/my/sweep/output/path/4bfa7f7c027a74878ce6cd0f6b09b0b4",
    "save_model_every_checkpoint": false,
    "seed": 2076257855,
    "skip_model_save": false,
    "steps": null,
    "task": "domain_generalization",
    "test_envs": [
      1
    ],
    "trial_seed": 0,
    "uda_holdout_fraction": 0
  },
  "env0_in_acc": 0.7466124661246613,
  "env0_out_acc": 0.717391304347826,
  "env1_in_acc": 0.4423305588585018,
  "env1_out_acc": 0.4095238095238095,
  "env2_in_acc": 0.9298831385642737,
  "env2_out_acc": 0.8926174496644296,
  "env3_in_acc": 0.5463552724699221,
  "env3_out_acc": 0.47875354107648727,
  "epoch": 325.54257095158596,
  "hparams": {
    "batch_size": 39,
    "class_balanced": false,
    "data_augmentation": true,
    "dinov2": false,
    "freeze_bn": false,
    "irm_lambda": 65030.52212762795,
    "irm_penalty_anneal_iters": 2,
    "lars": false,
    "linear_steps": 500,
    "lr": 2.7028930742148706e-05,
    "nonlinear_classifier": false,
    "resnet18": false,
    "resnet50_augmix": true,
    "resnet_dropout": 0.5,
    "vit": false,
    "vit_attn_tune": false,
    "vit_dropout": 0.1,
    "weight_decay": 0.00044832883881609976
  },
  "loss": -21626.141092834474,
  "mem_gb": 9.82351303100586,
  "nll": 2.4778925627470016,
  "penalty": -0.3325917995511554,
  "step": 5000,
  "step_time": 0.18920549511909485
}