Tensorboard Embedding visualizations

I was referring to demo code for embedding in pytorch. I am trying to create embedding of images for CIFAR 10, and unfortunately I am unable to so, my code block:

    dataiter = iter(testloader)
    img, lbl = dataiter.next()

    _out = net(Variable(img))
    _out = _out.data
    _, predicted = torch.max(_out,1)

    print('GroundTruth: ', ' '.join('%9s' % classes[lbl[j]] for j in range(9)))
    print('Predicted: ', ' '.join('%9s' % classes[predicted[j]]for j in range(9)))

    #make embeddings

    print("inputs: ",img.shape)
    print("outputs: ",_out.shape)
    print("targets: ",lbl.shape)

    writer.add_embedding(_out, metadata=lbl, label_img=img)

When I check the tensorboard, there is no logging. Kindly help me out.
Thanks in advance!

1 Like