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

tinygrad.utils to extra.utils fix in mnist_gan (#190)

Browse Source
This commit is contained in:
Marcel Bischoff
2020-12-12 23:52:36 -05:00
committed by GitHub
parent d901ef6b23
commit 6785614239
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/mnist_gan.py
View File

@@ -7,7 +7,7 @@ sys.path.append(os.getcwd())
sys.path.append(os.path.join(os.getcwd(), 'test'))
from tinygrad.tensor import Tensor, Function, register
from tinygrad.utils import get_parameters
from extra.utils import get_parameters
import tinygrad.optim as optim
from test_mnist import X_train, Y_train
from torchvision.utils import make_grid, save_image