Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Track TODOs #2

Open
3 of 10 tasks
nilsleh opened this issue Jan 9, 2025 · 1 comment
Open
3 of 10 tasks

Track TODOs #2

nilsleh opened this issue Jan 9, 2025 · 1 comment

Comments

@nilsleh
Copy link
Collaborator

nilsleh commented Jan 9, 2025

Just an issue to track some maybe nice to have TODOs:

  • make a script that automatically downloads (or shows how to download) the correct FM pretrained weights from HF, also take care of the torch.hub that they might not fill up the home directory
  • In the config.py file the model path has to be part of another env variable, because users will have them in different locations and shouldn't need to edit them, actually change again and use a url download to remove hf_upload
  • add F1 Score as accuracy metric which is more informative than just accuracy
  • Up to discussion, whether config.py as opposed to one large file, could be split into yaml files for example that can be merged with OmegaConf
  • Add CI linter like ruff that is run on PRs Run ruff and add it as CI linter style test #3
  • Other CI tests to ensure nothing breaks
  • Add simple sphinx documentation to include some tutorials
  • UperHead should have the same configuration across FMs for fair comparison?
  • Make additional base classes for classification and segmentation task to remove code duplication and ensure common configuration and metrics across FM encoders
  • add the option for full finetine or last layer (or decoder only) and for for more specialized cases expect the user to subclass and overwrite the params_to_optimize function
@ando-shah
Copy link
Contributor

  • add wandb support
  • merge ray configs with hydra config system

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

No branches or pull requests

2 participants