Skip to content
#

data-parallelism

Here are 22 public repositories matching this topic...

rsn870
rsn870 commented Aug 21, 2020

Hi ,

I have tried out both loss.backward() and model_engine.backward(loss) for my code. There are several subtle differences that I have observed , for one retain_graph = True does not work for model_engine.backward(loss) . This is creating a problem since buffers are not being retained every time I run the code for some reason.

Please look into this if you could.

Improve this page

Add a description, image, and links to the data-parallelism topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the data-parallelism topic, visit your repo's landing page and select "manage topics."

Learn more