refactor: set log level to separate out train args

foundation-model-stack / fms-hf-tuning

🚀 Collection of tuning recipes with HuggingFace SFTTrainer and PyTorch FSDP.

Apache License 2.0

28 stars 48 forks source link

refactor: set log level to separate out train args #314

Open anhuong opened 3 months ago

anhuong commented 3 months ago

Description of the change

Separate out python logging and setting log level with setting log level using train_args and altering train_args.log_level to match python logging level.

Related issue number

How to verify the PR

Was the PR tested

[ ] I have added >=1 unit test(s) for every new method I have added.
[ ] I have ensured all unit tests pass