Comment puis-je commencer à écrire le code de ma couche?

Question

J'ai vu que les chercheurs ajoutaient des fonctionnalités à la version originale de Caffe et utilisaient ces couches et fonctionnalités selon leurs besoins, puis ces versions sont partagées via Github. Si je ne me trompe pas, il y a deux façons: 1) en recompilant Caffe après avoir ajouté les versions C++ et Cuda des calques. 2) écrire un code python pour la fonctionnalité et l'appeler comme couche python dans Caffe.

Je souhaite ajouter un nouveau calque à Caffe en fonction de mon problème de recherche. Je ne sais vraiment pas à partir de quel point je devrais commencer à écrire la nouvelle couche et quelles étapes je devrais considérer.

Mes questions sont les suivantes:

1) Y at-il des documents ou tout matériel d'apprentissage que je peux l'utiliser pour écrire la couche?

2) De quelle manière les méthodes d'ajout d'une nouvelle couche mentionnées ci-dessus sont-elles préférées?

J'apprécie vraiment toute l'aide et des conseils

Merci beaucoup

Answer 1

Pour des fins de recherche, pour « jouer autour », il est généralement plus commode d'écrire une couche python: vous permet d'économiser l'agitation de la compilation, etc.
Vous trouverez un court tutoriel sur "Python" couche here. Par contre, si vous voulez de meilleures performances, vous devez écrire un code C++ natif pour votre couche.
Vous pouvez trouver une brève explication à ce sujet here.