enhancement: better model_save#13

Open

getorca wants to merge 1 commit into

midrender:mainfrom

getorca commented Dec 19, 2023

This pull request has 3 small changes to make saving the model easier:

sets a default value for _internal_call in save_model like in hf trainer https://github.com/huggingface/transformers/blob/c48787f347bd604f656c2cfff730e029c8f8c1fe/src/transformers/trainer.py#L2797C12-L2797C12
add arg parser for output_dir
calls save_model when training is complete


          enhancement: better model_save & output_dir arg

4d9d3fb

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

Labels

None yet