How to Solve NameError: name 'make_grid' is not defined -- torchvision

Solution: Import the 'make_grid' module

To Solve the error, add the following line to the top of your code.

from torchvision.utils import make_grid

 

For more information:
Python torchvision make_grid