• Nate Cibik's avatar
    Trainer - add cache clearing and the option for batched eval metrics computation (#28769) · df475bf8
    Nate Cibik authored
    * Added cache clearing for GPU efficiency.
    
    * Added cache clearing for GPU efficiency.
    
    * Added batch_eval_metrics capability
    
    * Ran make fixup
    
    * Fixed bug
    
    * Fixed whitespace issue
    
    * Fixed outdated condition
    
    * Updated docstrings with instructions for batch_eval_metrics. Updated end of dataloader logic
    
    * Added first version of batch_eval_metrics Trainer test
    
    * Fixed batch_eval_metrics Trainer tests for both eval and predict
    
    * Fixed batch_eval_metrics behavior for new Trainer variables
    
    * Fixed batch_eval_metrics Trainer tests
    
    * Ran fixup
    df475bf8
test_trainer.py 169 KB