Skip to content

Comments

Refactor ISIC fine-tuning code and restructure repo#3

Merged
sonnetx merged 2 commits intomainfrom
refactor_vicky
May 22, 2025
Merged

Refactor ISIC fine-tuning code and restructure repo#3
sonnetx merged 2 commits intomainfrom
refactor_vicky

Conversation

@Vicbi
Copy link
Collaborator

@Vicbi Vicbi commented May 22, 2025

Summary

This PR refactors the ViT fine-tuning pipeline for the ISIC 2019 dataset with the following improvements:

  • Cleaned and organized import statements.
  • Integrated argparse to eliminate hardcoded paths and make the script configurable (dataset name, output directory, resolutions, epochs).
  • Modularized metrics computation and GPU memory tracking.
  • Improved robustness of AUC calculation and result handling.
  • Clarified structure for better maintainability and extensibility.

@sonnetx sonnetx merged commit 23a90d6 into main May 22, 2025
2 of 7 checks passed
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.

2 participants