2026-05-01 20:23:11,903 | INFO | Starting training for model: efficientnetv2
2026-05-01 20:23:11,903 | INFO | Args: {'model_name': 'efficientnetv2', 'image_dir': '/root/NIH-ChestXray/data/images_all', 'csv_path': '/root/NIH-ChestXray/data/Data_Entry_2017_v2020.csv', 'train_split_file': '/root/NIH-ChestXray/data/train_val_list.txt', 'test_split_file': '/root/NIH-ChestXray/data/test_list.txt', 'output_dir': './outputs', 'epochs': 10, 'batch_size': 512, 'lr': 0.0003, 'weight_decay': 1e-05, 'image_size': 224, 'num_workers': 8, 'pos_weight': 3.0, 'patience': 3}
2026-05-01 20:23:12,179 | INFO | Using device: cuda
2026-05-01 20:23:12,207 | INFO | GPU: NVIDIA A100-SXM4-80GB
2026-05-01 20:23:12,916 | INFO | Train samples: 86524 | Val samples: 25596
2026-05-01 20:23:13,142 | INFO | Loading pretrained weights from Hugging Face hub (timm/tf_efficientnetv2_s.in21k_ft_in1k)
2026-05-01 20:23:13,280 | INFO | HTTP Request: HEAD https://huggingface.co/timm/tf_efficientnetv2_s.in21k_ft_in1k/resolve/main/model.safetensors "HTTP/1.1 302 Found"
2026-05-01 20:23:13,281 | WARNING | Warning: You are sending unauthenticated requests to the HF Hub. Please set a HF_TOKEN to enable higher rate limits and faster downloads.
2026-05-01 20:23:13,303 | INFO | HTTP Request: GET https://huggingface.co/api/models/timm/tf_efficientnetv2_s.in21k_ft_in1k/xet-read-token/ea9abc143ea2b9d8e1ec1de277bce02149b9cf0e "HTTP/1.1 200 OK"
2026-05-01 20:23:14,721 | INFO | [timm/tf_efficientnetv2_s.in21k_ft_in1k] Safe alternative available for 'pytorch_model.bin' (as 'model.safetensors'). Loading weights using safetensors.
2026-05-01 20:23:16,586 | INFO | Missing keys (classifier.weight, classifier.bias) discovered while loading pretrained weights. This is expected if model is being adapted.
2026-05-01 20:23:29,934 | INFO | Epoch 1 | Batch 0/169 | Loss: 2.5928
2026-05-01 20:25:19,851 | INFO | Epoch 1 | Batch 100/169 | Loss: 0.3750
2026-05-01 20:27:39,009 | INFO | Epoch 1/10 | Train Loss: 0.3935 | Val Loss: 0.4840 | Val Mean AUC: 0.7443 | Time: 262.2s
2026-05-01 20:27:39,010 | INFO | Per-class AUC: {
  "No Finding": 0.7129,
  "Atelectasis": 0.7299,
  "Cardiomegaly": 0.7744,
  "Effusion": 0.8051,
  "Infiltration": 0.6882,
  "Mass": 0.7129,
  "Nodule": 0.6834,
  "Pneumonia": 0.6675,
  "Pneumothorax": 0.8161,
  "Consolidation": 0.7096,
  "Edema": 0.8157,
  "Emphysema": 0.8227,
  "Fibrosis": 0.7493,
  "Pleural_Thickening": 0.6989,
  "Hernia": 0.7786
}
2026-05-01 20:27:39,538 | INFO | Saved best model with AUC: 0.7443
2026-05-01 20:27:50,532 | INFO | Epoch 2 | Batch 0/169 | Loss: 0.3251
2026-05-01 20:29:42,059 | INFO | Epoch 2 | Batch 100/169 | Loss: 0.3538
2026-05-01 20:32:01,567 | INFO | Epoch 2/10 | Train Loss: 0.3264 | Val Loss: 0.4660 | Val Mean AUC: 0.7841 | Time: 262.0s
2026-05-01 20:32:01,568 | INFO | Per-class AUC: {
  "No Finding": 0.7279,
  "Atelectasis": 0.7542,
  "Cardiomegaly": 0.866,
  "Effusion": 0.8155,
  "Infiltration": 0.7004,
  "Mass": 0.7828,
  "Nodule": 0.7213,
  "Pneumonia": 0.7117,
  "Pneumothorax": 0.8538,
  "Consolidation": 0.7336,
  "Edema": 0.8315,
  "Emphysema": 0.8889,
  "Fibrosis": 0.7977,
  "Pleural_Thickening": 0.7374,
  "Hernia": 0.8394
}
2026-05-01 20:32:02,527 | INFO | Saved best model with AUC: 0.7841
2026-05-01 20:32:13,602 | INFO | Epoch 3 | Batch 0/169 | Loss: 0.3199
2026-05-01 20:34:04,565 | INFO | Epoch 3 | Batch 100/169 | Loss: 0.3234
2026-05-01 20:36:22,404 | INFO | Epoch 3/10 | Train Loss: 0.3104 | Val Loss: 0.4542 | Val Mean AUC: 0.7939 | Time: 259.9s
2026-05-01 20:36:22,405 | INFO | Per-class AUC: {
  "No Finding": 0.7287,
  "Atelectasis": 0.7627,
  "Cardiomegaly": 0.8741,
  "Effusion": 0.8245,
  "Infiltration": 0.6975,
  "Mass": 0.8056,
  "Nodule": 0.7391,
  "Pneumonia": 0.7126,
  "Pneumothorax": 0.8531,
  "Consolidation": 0.7441,
  "Edema": 0.8316,
  "Emphysema": 0.9014,
  "Fibrosis": 0.8109,
  "Pleural_Thickening": 0.7593,
  "Hernia": 0.863
}
2026-05-01 20:36:23,174 | INFO | Saved best model with AUC: 0.7939
2026-05-01 20:36:34,020 | INFO | Epoch 4 | Batch 0/169 | Loss: 0.2990
2026-05-01 20:38:26,983 | INFO | Epoch 4 | Batch 100/169 | Loss: 0.2603
2026-05-01 20:40:46,888 | INFO | Epoch 4/10 | Train Loss: 0.2984 | Val Loss: 0.4543 | Val Mean AUC: 0.7977 | Time: 263.7s
2026-05-01 20:40:46,888 | INFO | Per-class AUC: {
  "No Finding": 0.7265,
  "Atelectasis": 0.7625,
  "Cardiomegaly": 0.8833,
  "Effusion": 0.8195,
  "Infiltration": 0.697,
  "Mass": 0.8045,
  "Nodule": 0.7442,
  "Pneumonia": 0.721,
  "Pneumothorax": 0.8645,
  "Consolidation": 0.7419,
  "Edema": 0.8455,
  "Emphysema": 0.9069,
  "Fibrosis": 0.8091,
  "Pleural_Thickening": 0.7692,
  "Hernia": 0.8703
}
2026-05-01 20:40:47,516 | INFO | Saved best model with AUC: 0.7977
2026-05-01 20:40:58,571 | INFO | Epoch 5 | Batch 0/169 | Loss: 0.2751
2026-05-01 20:42:49,535 | INFO | Epoch 5 | Batch 100/169 | Loss: 0.2734
2026-05-01 20:45:07,225 | INFO | Epoch 5/10 | Train Loss: 0.2850 | Val Loss: 0.4603 | Val Mean AUC: 0.7987 | Time: 259.7s
2026-05-01 20:45:07,225 | INFO | Per-class AUC: {
  "No Finding": 0.7298,
  "Atelectasis": 0.7628,
  "Cardiomegaly": 0.8798,
  "Effusion": 0.8228,
  "Infiltration": 0.6938,
  "Mass": 0.804,
  "Nodule": 0.7521,
  "Pneumonia": 0.7113,
  "Pneumothorax": 0.8545,
  "Consolidation": 0.7438,
  "Edema": 0.8461,
  "Emphysema": 0.9068,
  "Fibrosis": 0.8118,
  "Pleural_Thickening": 0.7681,
  "Hernia": 0.8933
}
2026-05-01 20:45:07,892 | INFO | Saved best model with AUC: 0.7987
2026-05-01 20:45:18,453 | INFO | Epoch 6 | Batch 0/169 | Loss: 0.2734
2026-05-01 20:47:09,922 | INFO | Epoch 6 | Batch 100/169 | Loss: 0.2664
2026-05-01 20:49:27,959 | INFO | Epoch 6/10 | Train Loss: 0.2674 | Val Loss: 0.4742 | Val Mean AUC: 0.7912 | Time: 260.1s
2026-05-01 20:49:27,960 | INFO | Per-class AUC: {
  "No Finding": 0.7234,
  "Atelectasis": 0.7558,
  "Cardiomegaly": 0.8693,
  "Effusion": 0.8189,
  "Infiltration": 0.681,
  "Mass": 0.8059,
  "Nodule": 0.7334,
  "Pneumonia": 0.7028,
  "Pneumothorax": 0.856,
  "Consolidation": 0.7258,
  "Edema": 0.8386,
  "Emphysema": 0.8978,
  "Fibrosis": 0.8111,
  "Pleural_Thickening": 0.7656,
  "Hernia": 0.8826
}
2026-05-01 20:49:28,289 | INFO | EarlyStopping: no improvement for 1/3 epochs
2026-05-01 20:49:39,172 | INFO | Epoch 7 | Batch 0/169 | Loss: 0.2397
2026-05-01 20:51:31,979 | INFO | Epoch 7 | Batch 100/169 | Loss: 0.2511
2026-05-01 20:53:50,933 | INFO | Epoch 7/10 | Train Loss: 0.2443 | Val Loss: 0.5064 | Val Mean AUC: 0.7816 | Time: 262.6s
2026-05-01 20:53:50,934 | INFO | Per-class AUC: {
  "No Finding": 0.7049,
  "Atelectasis": 0.7456,
  "Cardiomegaly": 0.8651,
  "Effusion": 0.8085,
  "Infiltration": 0.6687,
  "Mass": 0.7916,
  "Nodule": 0.734,
  "Pneumonia": 0.6827,
  "Pneumothorax": 0.8417,
  "Consolidation": 0.7197,
  "Edema": 0.833,
  "Emphysema": 0.8941,
  "Fibrosis": 0.8087,
  "Pleural_Thickening": 0.7584,
  "Hernia": 0.8671
}
2026-05-01 20:53:51,327 | INFO | EarlyStopping: no improvement for 2/3 epochs
2026-05-01 20:54:01,933 | INFO | Epoch 8 | Batch 0/169 | Loss: 0.2147
2026-05-01 20:55:53,388 | INFO | Epoch 8 | Batch 100/169 | Loss: 0.2219
2026-05-01 20:58:11,478 | INFO | Epoch 8/10 | Train Loss: 0.2162 | Val Loss: 0.5603 | Val Mean AUC: 0.7737 | Time: 260.2s
2026-05-01 20:58:11,479 | INFO | Per-class AUC: {
  "No Finding": 0.7023,
  "Atelectasis": 0.7307,
  "Cardiomegaly": 0.8571,
  "Effusion": 0.8029,
  "Infiltration": 0.6714,
  "Mass": 0.7837,
  "Nodule": 0.729,
  "Pneumonia": 0.6607,
  "Pneumothorax": 0.827,
  "Consolidation": 0.7057,
  "Edema": 0.8182,
  "Emphysema": 0.8911,
  "Fibrosis": 0.8014,
  "Pleural_Thickening": 0.746,
  "Hernia": 0.8788
}
2026-05-01 20:58:11,877 | INFO | EarlyStopping: no improvement for 3/3 epochs
2026-05-01 20:58:11,877 | INFO | Early stopping triggered at epoch 8. Best AUC: 0.7987
2026-05-01 20:58:11,877 | INFO | Training complete. Best Val AUC: 0.7987