Skip to content

完成 TinyInfiniTrain 全部作业与 A100 完整训练实验 #11

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

Open
pikaxinge wants to merge 2 commits into
base: master
Choose a base branch
from
Open

完成 TinyInfiniTrain 全部作业与 A100 完整训练实验 #11

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
fe7ef78
Complete TinyInfiniTrain assignment
pikaxinge Aug 12, 2026
6171272
Expand assignment report details
pikaxinge Aug 12, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,9 @@ add_library(example_gpt2 STATIC
)
target_link_libraries(example_gpt2 infini_train)

add_executable(train_experiment example/gpt2/train_experiment.cc)
target_link_libraries(train_experiment example_gpt2 infini_train)

function(build_test files)
# Non-recursive glob for skip failed tests
file(GLOB TEST_SOURCES ${files})
Expand All @@ -102,4 +105,4 @@ if(BUILD_TEST)
build_test(test/optimizer/test_adam_cuda.cc)
endif()
endif()
endif()
endif()
3 changes: 3 additions & 0 deletions Makefile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,3 +16,6 @@ clean:
test-cpp:
@echo
cd build/$(TYPE) && make test

train-experiment: build
./build/$(TYPE)/train_experiment $(ARGS)
9 changes: 8 additions & 1 deletion README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,4 +6,11 @@

[训练营作业介绍文档](docs/训练营作业介绍.md)

[训练营作业报告模版](docs/TinyInfiniTrain%20作业报告.md)
[训练营作业报告模版](docs/TinyInfiniTrain%20作业报告.md)

## 作业交付

- 六项作业实现及原有 8 个公开测试均已完成，测试文件未修改。
- [完整作业报告](docs/TinyInfiniTrain%20作业报告.md)
- [A100 完整训练指标](docs/experiments/a100-full-seed1337/summary.json)
- [A100 学习率对比](docs/experiments/a100-lr-sweep-seed1337/comparison.csv)
Loading