Skip to content

Commit 20f9c46

Browse files
committed
use integration test as the badge shown on the homepage
ghstack-source-id: 2674375b73e6c79db89716992da2e7c1f7c23a29 Pull Request resolved: #373
1 parent 1ceaa4e commit 20f9c46

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
[![GPU Integration Test](https://github.com/pytorch/torchtitan/actions/workflows/unit_test_4gpu.yaml/badge.svg?branch=main)](https://github.com/pytorch/torchtitan/actions/workflows/unit_test_4gpu.yaml)
1+
[![GPU Integration Test](https://github.com/pytorch/torchtitan/actions/workflows/integration_test_periodic.yaml/badge.svg?branch=main)](https://github.com/pytorch/torchtitan/actions/workflows/integration_test_periodic.yaml)
22

33
# torchtitan
44

train_configs/debug_model.toml

+1-2
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
[job]
44
dump_folder = "./outputs"
55
description = "Llama 3 debug training"
6-
# TODO: turn this back on once ci have tokenizer
76
use_for_integration_test = true
87

98
[profiling]
@@ -50,7 +49,7 @@ interval_type = "steps"
5049
interval = 5
5150
model_weights_only = false
5251
export_dtype = "float32"
53-
async_mode = "disabled" # ["disabled", "async", "async_with_pinned_mem"]
52+
async_mode = "disabled" # ["disabled", "async", "async_with_pinned_mem"]
5453

5554
[activation_checkpoint]
5655
mode = 'selective' # ['none', 'selective', 'full']

0 commit comments

Comments
 (0)