Skip to content

Howto include permutohedral op in MXNet? #1

@MarvinTeichmann

Description

@MarvinTeichmann

Hi Eric,

thanks for sharing you permutohedral lattice implementation. It looks like amazing work. I am trying to implement CRFasRNN into MxNet and would like to use your permutohedral cuda implementation.

I have tried cloning this repository to $MXNET/plugin/permutohedral, building mxnet from source and run test.py. However I am getting the error:

Traceback (most recent call last):
  File "test.py", line 178, in <module>
    sym = mx.sym.Permutohedral(data=mx.sym.Variable('val'), pos=mx.sym.Variable('pos'), normalize=norm)
AttributeError: 'module' object has no attribute 'Permutohedral'

I am using mxnet Version 0.9.5 under Linux. Is you implementation compatible with the 0.9 way of defining ops? What steps should I do to make test.py work?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions