Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

对Pytorch 添加多卡训练支持 #55

Open
wants to merge 6 commits into
base: master
Choose a base branch
from
Open

对Pytorch 添加多卡训练支持 #55

wants to merge 6 commits into from

Commits on Apr 19, 2020

  1. model.forward().shape(0) 应该是batch_size，不然多显卡训练很难完成 .Make sure model.f… 

    …orward().shape(0) is batch_size otherwise it's difficult to change code for multipgpu
    @cyy53589
    cyy53589 committed Apr 19, 2020
    Configuration menu
    Copy the full SHA
    40e7e4d View commit details
    Browse the repository at this point in the history

  2. pytorch 添加多卡训练，修改识别大写字母为识别大小+小写字母. Add multi-gpu training code. Chang… 

    …e recognizing uppercase letter to uppercase + lowercase
    @cyy53589
    cyy53589 committed Apr 19, 2020
    Configuration menu
    Copy the full SHA
    59e3f95 View commit details
    Browse the repository at this point in the history

  3. work 0->12

    @cyy53589
    cyy53589 committed Apr 19, 2020
    Configuration menu
    Copy the full SHA
    2f2616a View commit details
    Browse the repository at this point in the history

  4. Update ctc_pytorch_multigpu.py 

    在forward() 里面，LSTM之前的shape跟作者保持一致，不然训练不出来。
    @cyy53589
    cyy53589 authored Apr 19, 2020
    Configuration menu
    Copy the full SHA
    bd891c2 View commit details
    Browse the repository at this point in the history

  5. Update ctc_pytorch_multigpu.py

    @cyy53589
    cyy53589 authored Apr 19, 2020
    Configuration menu
    Copy the full SHA
    6d87f20 View commit details
    Browse the repository at this point in the history

  6. bugs fix

    @cyy53589
    cyy53589 committed Apr 19, 2020
    Configuration menu
    Copy the full SHA
    02033ad View commit details
    Browse the repository at this point in the history