test commit 3

gkielian · Nov 7, 2023 · 1219a23 · 1219a23
1 parent b18310e
commit 1219a23
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/.github/workflows/shakespeare-char-test.yml b/.github/workflows/shakespeare-char-test.yml
@@ -15,5 +15,13 @@ jobs:
       - name: List files in the repository
         run: |
           ls ${{ github.workspace }}
+      - name: install deps for cpu inference
+        run: |
+          python3 -m pip install --upgrade pip
+          pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+          pip3 install torch numpy transformers datasets tiktoken wandb tensorboard tqdm
+      - name: run small network for cpu inference test
+        run: |
+          python3 train.py --device=cpu --eval_iters=1 --log_interval=1 --block_size=2 --batch_size=2 --n_layer=2 --n_head=2 --n_embd=16 --max_iters=2 --lr_decay_iters=2 --dropout=0.0
       - run: echo "🍏 This job's status is ${{ job.status }}."