Manually invoke update of `data.json`.

[pascal-mueller]

Hello,

I would like to look at the computational graph at different stages of it. For specific reason I have to use autograd.grad which seems doesn't update data.json at all. It seems that torchexplorer uses handle_step() to update the data.json which in turn is called in these two hooks:

    module.register_forward_pre_hook(pre_forward_hook)
    module.register_full_backward_hook(post_backward_hook)

I think those two hooks are never called when you use autograd.grad. I'd like to be able to check out the computational graph at any time, so I was wondering if there's a way to manually update data.json?