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

Update examples to new API (#205)

Browse Source
This commit is contained in:
Liam
2020-12-19 18:04:12 +01:00
committed by GitHub
parent 3c9150a9f4
commit 075cf4bc02
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/mnist_gan.py
View File

@@ -65,7 +65,7 @@ if __name__ == "__main__":
ds_noise = Tensor(np.random.randn(64,128).astype(np.float32), gpu=GPU, requires_grad=False)
n_steps = int(train_data_size/batch_size)
if GPU:
[x.cuda_() for x in generator_params+discriminator_params]
[x.gpu_() for x in generator_params+discriminator_params]
# optimizers
optim_g = optim.Adam(generator_params,lr=0.0002, b1=0.5) # 0.0002 for equilibrium!
optim_d = optim.Adam(discriminator_params,lr=0.0002, b1=0.5)