search

moskewcz / boda

Boda: A C++ Framework for Efficient Experiments in Computer Vision
Other
63 stars 12 forks source link

Add ResNet layer types: Scale, BatchNorm, Eltwise? #11

Closed forresti closed 8 years ago

forresti commented 8 years ago

I'm experimenting with the ResNet prototxt files from MSR.

At this stage, I'm just trying to use Boda to count the number of FLOPS in the ResNet-50 architecture. After I added a proper data layer, boda cnet-ana accepts the ResNet-50 prototxt file. But, Boda isn't aware of these layer types: BatchNorm, Eltwise, and Scale. Would it be possible to implement function signatures in Boda for these 3 layer types, plus just enough of a stub to make FLOP counting feasible?

Here's a partial stack trace from Boda when I feed this architecture into cnet_ana:

warning: ignoring layer with lp.type()=Eltwise
warning: ignoring layer with lp.type()=Scale
warning: ignoring layer with lp.type()=BatchNorm
error: calc_support_info(): needed input support info for node not set. node name: res2a
forresti commented 8 years ago

Thanks!

I'm trying to bring up your current Boda version on my machine, so that I can use this improvement.

In the branch I've been using locally, my latest commit from Boda-master is: aef92df from Feb 2016.

Have the locations of config files (e.g. boda_cfg.xml) changed since February?