config.json

{
  "model_type": "GPT2WithCLSHead",
  "architecture": "GPT2WithCLSHead",
  "vocab_size": 516,
  "n_embd": 384,
  "n_layer": 6,
  "n_head": 8,
  "n_positions": 1024,
  "num_classes": 10,
  "use_attention_pooling": true,
  "classifier_dropout": 0.1,
  "task_type": "image_classification",
  "best_epoch": 10,
  "best_val_accuracy": 0.11057692307692307,
  "best_val_loss": 2.3339636142437277,
  "training_config": {
    "batch_size": 16,
    "num_epochs": 20,
    "learning_rate": 3e-05,
    "weight_decay": 0.01,
    "gradient_clip": 1.0,
    "output_dir": "outputs/v7_models",
    "checkpoint_dir": "outputs/v7_checkpoints",
    "log_dir": "outputs/v7_logs"
  },
  "saved_at": "2025-09-01T13:10:57.172343"
}