Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
allamo_pytorch
Commits
f828a7cd
Commit
f828a7cd
authored
Nov 27, 2024
by
chenzk
Browse files
v1.0.4
parent
c2faed49
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
README.md
README.md
+1
-1
No files found.
README.md
View file @
f828a7cd
...
...
@@ -134,7 +134,7 @@ cd /home/allamo
mkdir /home/data/out-allamo-1B
python train.py --config="./train_configs/train_1B.json"# 或sh train.sh
# 其它功能正在优化中
# 其它功能正在优化中
,欢迎共同优化和拓展。
```
若希望用原始规模的Llama进行预训练,可修改
`train_configs/train_1B.json`
中的参数获得新模型配置文件,但大规模模型可能显存超过设备容量,需自行调试合适规模,如
`Meta-Llama-3.1-8B`
的模型参数可参考(本项目要获得Llama系列模型,dropout都设置为0。):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment