mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

tinygrad.utils -> extra.utils

Browse source
This commit is contained in:
George Hotz 2020-12-12 15:26:07 -08:00
commit 1d10559d1d
7 changed files with 6 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/efficientnet.py
View file

@ -10,7 +10,7 @@ import time
import numpy as np
np.set_printoptions(suppress=True)
from tinygrad.tensor import Tensor
from tinygrad.utils import fetch, get_parameters
from extra.utils import fetch, get_parameters
from extra.efficientnet import EfficientNet
def infer(model, img):

2
examples/serious_mnist.py
View file

@ -7,8 +7,8 @@ sys.path.append(os.path.join(os.getcwd(), 'test'))
from tinygrad.tensor import Tensor
from tinygrad.nn import BatchNorm2D
from tinygrad.utils import get_parameters
import tinygrad.optim as optim
from extra.utils import get_parameters
# TODO: abstract this generic trainer out of the test
from test_mnist import train as train_on_mnist

2
examples/train_efficientnet.py
View file

@ -3,7 +3,7 @@ import time
import numpy as np
from extra.efficientnet import EfficientNet
from tinygrad.tensor import Tensor
from tinygrad.utils import get_parameters, fetch
from extra.utils import get_parameters, fetch
from tqdm import trange
import tinygrad.optim as optim
import io

2
extra/efficientnet.py
View file

@ -1,8 +1,8 @@
import math
import numpy as np
from tinygrad.tensor import Tensor
from tinygrad.utils import fetch
from tinygrad.nn import BatchNorm2D
from extra.utils import fetch
USE_TORCH = False

2
test/test_mnist.py
View file

@ -3,8 +3,8 @@ import os
import unittest
import numpy as np
from tinygrad.tensor import Tensor, GPU
from tinygrad.utils import fetch, get_parameters
import tinygrad.optim as optim
from extra.utils import fetch, get_parameters
from tqdm import trange
# mnist loader

2
test/test_optim.py
View file

@ -3,7 +3,7 @@ import torch
import unittest
from tinygrad.tensor import Tensor, GPU
from tinygrad.optim import Adam, SGD, RMSprop
from tinygrad.utils import get_parameters
from extra.utils import get_parameters
x_init = np.random.randn(1,3).astype(np.float32)
W_init = np.random.randn(3,3).astype(np.float32)

1
tinygrad/__init__.py
View file

@ -1,4 +1,3 @@
import tinygrad.optim
import tinygrad.tensor
import tinygrad.utils
import tinygrad.nn