Running the code with the recent versions of the libraries gives me an error `AttributeError: 'function' object has no attribute 'Variable'` in DeepLKBatch on the lines like `if isinstance(p, torch.autograd.variable.Variable):` The problem is fixed by converting the reference to `torch.autograd.Variable`. See https://github.com/lanpa/tensorboardX/issues/100 Even if the original authors are not going to fix it, I hope the information will help those who try to use their code.