Can you print out the shape of images that you pass into the tensorboard add graph function. Ex:
    images,labels=next(iter(train_loader))
    print(images.shape)
    grid=torchvision.utils.make_grid(images)
    
    tensorboard=SummaryWriter(comment=comment)
    tensorboard.add_image("images",grid)
    tensorboard.add_graph(net, images)

what is the shape?