Running for a single image using _visualize.py

[vatsala96]

Hey thanks for sharing the code..
Running this code ..I have got the model and in rel_model.py getitem when I pass the image tensor to the model ..
Error is forward () takes at least 4 argument (2 given)
In nn.model.py

[rowanz]

hi there,

sorry for taking so long to respond. the visualization code isn't supported with the current version of the code. if you managed to get it to work i'd love a PR! thanks,

Rowan