Solved by Neda in post #2 
              
              
                This is my snippet to display the loss curve in visdom which is working fine. 
    vis = visdom.Visdom()
    loss_window = vis.line(X=torch.zeros((1,)).cpu(),
                           Y=torch.zeros((1)).cpu(),
                           opts=dict(xlabel='epoch',
                                   …