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

Refactor the code in train.py #213

Merged

pengzhendong merged 1 commit into main from sqli-dev

Mar 27, 2024

Collaborator

Shengqiang-Li commented Mar 19, 2024 •

edited

Loading

The initialization consists of seven parts: (1) Get the dataset and data loader; (2) Get the tts model; (3) Dispatch the model from cpu to gpu; (4) Get the optimizer; (5) Load the checkpoint; (6) Get the scheduler; (7) Get the tensorboard summary.
Set the device to local_rank instead of rank;


          Refactor the code in train.py

6463c8e

Shengqiang-Li requested review from pengzhendong and robin1001

March 19, 2024 10:35

pengzhendong approved these changes

View reviewed changes

pengzhendong merged commit 4dd2794 into main

3 checks passed

pengzhendong deleted the sqli-dev branch

March 27, 2024 07:47

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet