checkpoints/mobius_base_tiny_imagenet/20260110_132436/config.json

{
  "model": {
    "in_chans": 3,
    "num_classes": 200,
    "channels": [
      128,
      256,
      512,
      768,
      1024
    ],
    "depths": [
      2,
      2,
      2,
      2,
      2
    ],
    "scale_range": [
      0.25,
      2.75
    ],
    "use_integrator": true,
    "total_layers": 10,
    "num_stages": 5
  },
  "training": {
    "epochs": 200,
    "lr": 0.0003,
    "batch_size": 128,
    "optimizer": "AdamW",
    "weight_decay": 0.05,
    "scheduler": "CosineAnnealingLR",
    "total_params": 33709338
  },
  "timestamp": "20260110_132436",
  "variant_name": "mobius_base",
  "dataset_name": "tiny_imagenet"
}