Skip to content

Refactor for HF Trainer#1

Open
DavidC001 wants to merge 1 commit intomainfrom
codex/convert-to-use-huggingface-trainer-class
Open

Refactor for HF Trainer#1
DavidC001 wants to merge 1 commit intomainfrom
codex/convert-to-use-huggingface-trainer-class

Conversation

@DavidC001
Copy link
Copy Markdown
Owner

Summary

  • add TrainingConfig dataclass and extend project configuration
  • include training parameters in config.example.yaml
  • refactor train_trainer.py to read hyperparameters from config

Testing

  • python -m py_compile train_trainer.py trainer_hf.py llava_finetune/model.py configuration.py
  • pytest -q

https://chatgpt.com/codex/tasks/task_e_6841b8b0f8148326ab09b1db5b2f14c6

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

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant