github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

tinytqdm.set_description and tinytrange (#5101)

Browse Source
This commit is contained in:
chenyu
2024-06-22 14:45:06 -04:00
committed by GitHub
parent 8080298739
commit e356807696
13 changed files with 43 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/beautiful_mnist_multigpu.py
View File

@@ -1,9 +1,8 @@
# model based off https://towardsdatascience.com/going-beyond-99-mnist-handwritten-digits-recognition-cfff96337392
from typing import List, Callable
from tinygrad import Tensor, TinyJit, nn, GlobalCounters, Device
from tinygrad.helpers import getenv, colored
from tinygrad.helpers import getenv, colored, trange
from extra.datasets import fetch_mnist
from tqdm import trange
GPUS = [f'{Device.DEFAULT}:{i}' for i in range(getenv("GPUS", 2))]