Skip to content

Commit 2944a9d

Browse files
framoncgsoumith
authored andcommitted
Change run_python_examples.sh to run word_language_model/generate.py after each call to word_language_model/main.py
1 parent 6595d7b commit 2944a9d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

run_python_examples.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,10 +155,11 @@ function vision_transformer() {
155155

156156
function word_language_model() {
157157
uv run main.py --epochs 1 --dry-run $CUDA_FLAG --mps || error "word_language_model failed"
158+
uv run generate.py $CUDA_FLAG --mps || error "word_language_model generate failed"
158159
for model in "RNN_TANH" "RNN_RELU" "LSTM" "GRU" "Transformer"; do
159160
uv run main.py --model $model --epochs 1 --dry-run $CUDA_FLAG --mps || error "word_language_model failed"
161+
uv run generate.py $CUDA_FLAG --mps || error "word_language_model generate failed"
160162
done
161-
uv run generate.py $CUDA_FLAG --mps || error "word_language_model generate failed"
162163
}
163164

164165
function gcn() {

0 commit comments

Comments
 (0)