Unexpected keyword argument 'bf16' : Modified train.py to handle exceptions appropriately

openai / weak-to-strong

MIT License

2.5k stars 305 forks source link

Unexpected keyword argument 'bf16' : Modified train.py to handle exceptions appropriately #20

Closed srivhash closed 9 months ago

srivhash commented 10 months ago

Added a custom function that checks if the enlisted parameters within kwargs are valid. If the type error is thrown, then the function ignores that specific parameter and checks for the other parameters.

srivhash commented 10 months ago

Added the requested changes

WuTheFWasThat commented 9 months ago

using https://github.com/openai/weak-to-strong/pull/22 instead