"torchvision/csrc/io/image/cpu/readjpeg_impl.cpp" did not exist on "7d831a2f9b3ebab9eb8e5c899cf70b103ad6908a"
  • Lingfan Yu's avatar
    [Model] Support Multi-GPU for Transformer model (#356) · 29dd22e6
    Lingfan Yu authored
    * multi-process version of transformer
    
    * lots of fix
    
    * fix bugs and accum gradients for multiple batches
    
    * many fixes
    
    * minor
    
    * upd
    
    * set torch device
    
    * fix bugs
    
    * fix and minor
    
    * comments and clean up
    
    * uncomment viz code
    29dd22e6
fields.py 1.94 KB