Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

E

emoUS-public
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Iterations
- Wiki
- Requirements
- Environments
- Terraform modules
- Incidents

Snippets Groups Projects

76c1d32e

Commit 76c1d32e authored 2 years ago by zqwerty

Downloads
- Patches
- Plain Diff

update get token loss method for multi-gpu inference

parent 46df55a3

No related branches found

No related tags found

No related merge requests found

Changes 4

Show whitespace changes

Inline Side-by-side

Showing

convlab/base_models/gpt/create_data.py 1 addition, 1 deletion

convlab/base_models/gpt/create_data.py
convlab/base_models/gpt/keyword_extraction/get_token_loss.sh 35 additions, 0 deletions

convlab/base_models/gpt/keyword_extraction/get_token_loss.sh
convlab/base_models/gpt/run_clm.py 18 additions, 55 deletions

convlab/base_models/gpt/run_clm.py
convlab/base_models/gpt/trainer.py 243 additions, 0 deletions

convlab/base_models/gpt/trainer.py

with 297 additions and 56 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment