Open midhunharikumar opened 3 years ago
Values for canvas size > 32 fails at loss.backward() step running on GPU ( V100 ).
loss.backward()
Settings:
canvas_width = 240 canvas_height = 240 num_paths = 10 num_segments = 10
Notebook freezes and nvidia-smi shows gpu memory usage drop to 0% . IPython kernel is reset.
nvidia-smi
Values for canvas size > 32 fails at
loss.backward()
step running on GPU ( V100 ).Settings:
canvas_width = 240 canvas_height = 240 num_paths = 10 num_segments = 10
Notebook freezes and
nvidia-smi
shows gpu memory usage drop to 0% . IPython kernel is reset.