models_gan_vs_diffusion/config.json

{
  "task": "gan_vs_diffusion",
  "gan_dataset": "kmk29774/my-gan-training-images",
  "gan_config": null,
  "diffusion_dataset": "lesc-unifi/dragon",
  "diffusion_config": "ExtraLarge",
  "image_size": 224,
  "batch_size": 32,
  "epochs": 50,
  "lr": 0.0003,
  "weight_decay": 0.0001,
  "backbone": "resnet18",
  "label_to_id": {
    "gan": 0,
    "diffusion": 1
  },
  "id_to_label": {
    "0": "gan",
    "1": "diffusion"
  },
  "val_ratio": 0.1,
  "test_ratio": 0.1,
  "max_samples_per_class": 4500,
  "seed": 42,
  "best_val_f1": 1.0,
  "test_metrics": {
    "loss": 6.70997362400764e-06,
    "accuracy": 1.0,
    "f1": 1.0,
    "auc": 1.0,
    "precision_gan": 1.0,
    "recall_gan": 1.0,
    "precision_diffusion": 1.0,
    "recall_diffusion": 1.0
  },
  "weights_path": "models_gan_vs_diffusion\\best_model_weights.pt"
}