how do I get the neural network name inside hook function, so that I see output of different neural network under a different name in tensorboard
writer = SummaryWriter()
def hook(module, input, output):
    # print(input[0].shape, output.shape)
    grid = torchvision.utils.make_grid(output[0][0])
    # print(grid.shape)
    
    # how to change conv_features to neural network name?
    # so it is model.modelone.encoder for a, model.modeltwo.encoder for b
    writer.add_image('conv_features', grid, 0) 

a = model.modelone.encoder.register_forward_hook(hook)
b = model.modeltwo.encoder.register_forward_hook(hook)

module.__class__, module.__class__.__name__, module._get_name do not give neural network name.