• Glenn Jocher's avatar
    AutoBatch checks against failed solutions (#8159) · 6e466177
    Glenn Jocher 提交于
    * AutoBatch checks against failed solutions
    
    @kalenmike this is a simple improvement to AutoBatch to verify that returned solutions have not already failed, i.e. return batch-size 8 when 8 already produced CUDA out of memory.
    
    This is a halfway fix until I can implement a 'final solution' that will actively verify the solved-for batch size rather than passively assume it works.
    
    * Update autobatch.py
    
    * Update autobatch.py
    6e466177
autobatch.py 2.5 KB