2017-08-22 5 views
2

Je veux essayer de nouvelles fonctions d'activation et j'ai réussi à créer des opérateurs personnalisés en C++ en plus du dernier code source de mxnet. Mais ce que j'aime vraiment, c'est construire une bibliothèque séparée (fichier .so dans ubuntu) uniquement pour mes opérateurs personnalisés afin que je puisse laisser le code source de mxnet et sa bibliothèque tels quels et je n'ai pas à me soucier de fusionner mon code Code source mxnet chaque fois que j'ai besoin d'utiliser les nouvelles versions de mxnet. Je comprends que si cela fonctionne, j'ai besoin de libmxnet.so et mycustom.so pour utiliser mes nouvelles fonctions d'activation. Toute aide est appréciée. Merci d'avance.Comment construire des opérateurs personnalisés MXNET dans une bibliothèque/un paquet séparé avec C++?

+0

Une fonction d'activation utile pour votre réseau neuronal va probablement être utile pour d'autres. Pourquoi ne pas soumettre une demande d'extraction et la fusionner dans le dépôt? –

+0

Je peux penser à plusieurs raisons pour lesquelles il serait difficile de déployer des opérateurs personnalisés dans la bibliothèque officielle: le code n'est peut-être pas prêt pour la production, il pourrait ne pas être assez utile pour justifier la distribution principale. émis sous une licence en conflit (par exemple GPL), etc. –

+0

Il existe un problème: https://github.com/apache/incubator-mxnet/issues/9547 – tauran

Répondre

1

Comme noté dans le commentaire par @tauran, un problème a été ouvert dans github.