github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix float64 warning on training

Browse Source
This commit is contained in:
George Hotz
2021-10-30 20:07:31 -07:00
parent b0f14b4af8
commit c162e748f5
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_mnist.py
View File

@@ -5,7 +5,7 @@ import numpy as np
from tinygrad.tensor import Tensor
import tinygrad.optim as optim
from extra.training import train, evaluate
from extra.utils import fetch, get_parameters
from extra.utils import get_parameters
from datasets import fetch_mnist
# load the mnist dataset