"include/vscode:/vscode.git/clone" did not exist on "5625dd333a621932c469e2261fd9f0c8687378f8"
  • Kun Lin's avatar
    [shardformer] support vision transformer (#4096) · 8af29ee4
    Kun Lin authored
    * first v of vit shardformer
    
    * keep vit
    
    * update
    
    * vit shard add vitattention vitlayer
    
    * update num head shard para
    
    * finish test for vit
    
    * add new_model_class & postprocess
    
    * add vit readme
    
    * delete old files & fix the conflict
    
    * fix sth
    8af29ee4
test_device_mesh.py 2.97 KB