MCPcopy Index your code
hub / github.com/InternLM/InternLM / intern_moss_example.py

File intern_moss_example.py

tools/transformers/intern_moss_example.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import torch
2from torch.utils.data import DataLoader
3from peft import get_peft_model, LoraConfig, TaskType
4from transformers import get_linear_schedule_with_warmup

Callers

nothing calls this directly

Calls 8

get_datasetFunction · 0.85
collate_fnFunction · 0.85
writeMethod · 0.80
trainMethod · 0.45
backwardMethod · 0.45
stepMethod · 0.45
zero_gradMethod · 0.45
generateMethod · 0.45

Tested by

no test coverage detected