Work done in Microsoft AI Research.

Run the code with:-

python main.py

Run configuration can be modified in:-

config_small.gin

For Grad-cam visualizations, load the model you'd like to visualize on line 228 in grad-cam.py and run with:-

python grad-cam.py --use-cuda

The grad visualizations will be written to a folder outputs.