"vscode:/vscode.git/clone" did not exist on "9f21782542d13cd05c2c9c257a7ad0ea87b97cc0"
  • mcarilli's avatar
    WIP: Handle arbitrary combinations of optimizers/models/losses (#232) · 3f87614f
    mcarilli authored
    * Refactor to allow more flexible treatment of multiple optimizers/models/losses
    
    * Adding _process_optimizers.py
    
    * Created L0 tests (now passing).
    
    * fix: minor print typo (#234)
    
    * make L1 results easier to read
    
    * L0 multiple model/optimizer/loss test fleshed out
    
    * Adding test that master params remain synced across distributed processes
    
    * Docstring updates
    
    * Docstring updates
    3f87614f
multi_tensor_axpby_kernel.cu 3.21 KB