• HELSON's avatar
    [zero] fix error for BEiT models (#2169) · 24586599
    HELSON authored
    * [zero] fix error for BEiT models
    
    * [ColoParameter] add unpack operation for tuple arguments
    
    * fix bugs
    
    * fix chunkv2 unit testing
    
    * add assertion for gradient state
    24586599
test_chunkv2.py 4.21 KB