feat✨: 更新README和训练脚本,简化环境设置并移除不必要的代码
This commit is contained in:
parent
3b606da46f
commit
7d70d85c60
21
README.md
21
README.md
@ -3,23 +3,6 @@
|
|||||||
## Requirements
|
## Requirements
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
uv venv --python 3.11.7
|
uv sync
|
||||||
uv pip install torch==2.5.1+cu124 torchvision==0.20.1+cu124 torchaudio==2.5.1+cu124 --index-url https://download.pytorch.org/whl/cu124
|
uv sync --extra compile
|
||||||
uv pip install -r requirements.txt
|
|
||||||
```
|
|
||||||
|
|
||||||
## Code Structure
|
|
||||||
|
|
||||||
```bash
|
|
||||||
src
|
|
||||||
├── collatefn_library
|
|
||||||
├── configs accelerator configs
|
|
||||||
├── dataset_library
|
|
||||||
├── evaluation.py
|
|
||||||
├── evaluation.sh evaluation bash(TODO:update)
|
|
||||||
├── peft_library
|
|
||||||
├── todo.md
|
|
||||||
├── train.py
|
|
||||||
├── train.sh train bash
|
|
||||||
└── utils
|
|
||||||
```
|
```
|
||||||
|
@ -89,7 +89,6 @@ if __name__ == "__main__":
|
|||||||
accelerator.wait_for_everyone()
|
accelerator.wait_for_everyone()
|
||||||
|
|
||||||
model.eval()
|
model.eval()
|
||||||
accelerator = trainer.accelerator
|
|
||||||
|
|
||||||
from torch.utils.data import DataLoader
|
from torch.utils.data import DataLoader
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user