config.json

{
  "architectures": [
    "Model"
  ],
  "dataset": {
    "batch_size": 32,
    "correlation_strength": 0.95,
    "drop_last": false,
    "pin_memory": true
  },
  "max_epochs": 20,
  "metric_loss": true,
  "metric_loss_beta": 1.0,
  "metric_loss_hard": true,
  "metric_loss_margin": 0.1,
  "model_type": "resnet50",
  "optimizer": {
    "lr": 1e-05,
    "weight_decay": 1.0
  },
  "out_features": 6,
  "torch_dtype": "float32",
  "transformers_version": "4.38.2",
  "use_cuda": true
}