• Wenhao Chen's avatar
    [chat] remove naive strategy and split colossalai strategy (#4094) · edd75a59
    Wenhao Chen authored
    * feat: remove on_learn_epoch fn as not used
    
    * revert: add _on_learn_epoch fn
    
    * to: remove the use of NaiveStrategy
    
    * test: remove NaiveStrategy tests
    
    * feat: remove NaiveStrategy
    
    * style: modify comments and params
    
    * feat: split ColossalAIStrategy into LowLevelZeroStrategy and GeminiStrategy
    
    * fix: remove naive
    
    * fix: align with modified colossal strategy
    
    * fix: fix ddp _try_init_dist arg
    edd75a59
utils.py 5.95 KB