Skip to content

Fix eval metrics and wandb config for transformers (#192)#203

Open
DhruvrajSinhZala24 wants to merge 1 commit intoML4SCI:mainfrom
DhruvrajSinhZala24:fix/issue-192-eval-metrics
Open

Fix eval metrics and wandb config for transformers (#192)#203
DhruvrajSinhZala24 wants to merge 1 commit intoML4SCI:mainfrom
DhruvrajSinhZala24:fix/issue-192-eval-metrics

Conversation

@DhruvrajSinhZala24
Copy link
Copy Markdown

Description

  • Compute micro and macro AUROC plus per-class AUROCs in one pass; use an explicit softmax dim for ROC/confusion plots.
  • Add a --entity flag (defaults to WANDB_ENTITY or current user) in train/eval and remove the hardcoded W&B entity.
  • Improve training hygiene: seed-aware train/val split when --seed is set, log learning rate alongside loss, save the best checkpoint once per epoch, and simplify image-size selection; Model_IV option removed.

Testing

python3 -m py_compile DeepLense_Classification_Transformers_Archil_Srivastava/eval.py DeepLense_Classification_Transformers_Archil_Srivastava/train.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant