visdial icon indicating copy to clipboard operation
visdial copied to clipboard

Support for extracting conv layer / spatial features for ResNets (for attention-based models)

Open abhshkdz opened this issue 7 years ago • 1 comments
trafficstars

abhshkdz avatar Aug 11 '18 16:08 abhshkdz

Needs a command-line argument (like the VGG models) and corresponding model:remove() calls here: https://github.com/batra-mlp-lab/visdial/blob/master/data/prepro_img_resnet.lua#L37

abhshkdz avatar Aug 11 '18 16:08 abhshkdz