"...lm-evaluation-harness.git" did not exist on "9a64e6423d1bb4b10035a86cf0aa55749a7b5025"
  • atturaioe's avatar
    Add AnyPrecisionAdamW optimizer (#18961) · 84c9cc6d
    atturaioe authored
    * Add AnyPrecisionAdamW optimizer
    
    * Add optim_args argument to TrainingArgs
    
    * Add tests for AnyPrecisionOptimizer
    
    * Change AnyPrecisionAdam default params to float32
    
    * Move default_anyprecision_kwargs in trainer test
    
    * Rename AnyPrecisionAdamW
    84c9cc6d
test_trainer.py 103 KB