Skip to content

Instantly share code, notes, and snippets.

@EddieKro
EddieKro / prepare_vinvl_input.ipynb
Created April 29, 2021 08:00
VINVL inference: how to transform output from SG benchmark into VINVL input for a random image
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
from graphviz import Digraph
from torch.autograd import Variable
import torch
def make_dot(var, params=None):
if params is not None:
assert isinstance(params.values()[0], Variable)
param_map = {id(v): k for k, v in params.items()}