Solved by ssgosh in post #2 
              
              
                Hello, I might be a bit late, but you may use torchvision.utils.make_grid() to accomplish this. 
Example code: 
class TensorBoardHelper:

    def __init__(self, name=None):
        self.writer = SummaryWriter(name)


    def add_image_grid(self, images, tag, global_step):
        images = images.det…