Skip to content

Commit 1ad9802

Browse files
authored
Update dlnd_face_generation.ipynb
I've experienced CUDA memory leakage during training, the CUDA memory was increasing at the end of each epoch. This change has solved the issue for me.
1 parent 388f43c commit 1ad9802

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

project-face-generation/dlnd_face_generation.ipynb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -633,8 +633,10 @@
633633
" # this code assumes your generator is named G, feel free to change the name\n",
634634
" # generate and save sample, fake images\n",
635635
" G.eval() # for generating samples\n",
636-
" samples_z = G(fixed_z)\n",
637-
" samples.append(samples_z)\n",
636+
" with torch.no_grad():\n",
637+
" samples_z = G(fixed_z)\n",
638+
" samples_z = samples_z.detach().cpu()\n",
639+
" samples.append(samples_z)\n",
638640
" G.train() # back to training mode\n",
639641
"\n",
640642
" # Save training generator samples\n",

0 commit comments

Comments
 (0)