zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-07-16 23:41:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Updating GAN Code...

Browse Source
This commit is contained in:
Shubbair 2024-07-28 17:26:24 +03:00
parent 3716501e8d
commit 3e63cd93fe
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
gan/playground.ipynb
View File

@ -463,14 +463,6 @@
"for epoch in tqdm(range(200)):\n",
"\n",
" for real in batch_iterate(batch_size, train_images):\n",
" cur_batch_size = len(real)\n",
" # real = real.reshape(-1)\n",
" \n",
" # Flatten the batch of real images from the dataset\n",
" \n",
" # plt.imshow(real[0].reshape(28,28))\n",
" # print(len(real))\n",
" # break\n",
" \n",
" D_loss,D_grads = D_loss_grad(gen, disc, real, batch_size, z_dim)\n",
"\n",