• Jiarui Fang's avatar
    [zero] zero init context (#321) · de0468c7
    Jiarui Fang authored
    * add zero init context
    
    * add more flags for zero init context
    fix bug of repeated converting param to ShardedParamV2
    
    * polish code
    de0468c7
test_init_context.py 1.21 KB