• Glenn Jocher's avatar
    Add `autobatch` feature for best `batch-size` estimation (#5092) · ca19df5f
    Glenn Jocher 提交于
    * Autobatch
    
    * fix mem
    
    * fix mem2
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update train.py
    
    * print result
    
    * Cleanup print result
    
    * swap fix in call
    
    * to 64
    
    * use total
    
    * fix
    
    * fix
    
    * fix
    
    * fix
    
    * fix
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Cleanup printing
    
    * Update final printout
    
    * Update autobatch.py
    
    * Update autobatch.py
    
    * Update autobatch.py
    ca19df5f
autobatch.py 2.0 KB