so berechnen Sie eine net-FLOPs in der CNN
Möchte ich design ein convolutional neural network, die occupy-GPU-Ressourcen-nicht mehr als Alexnet.Ich will-FLOPs zu Messen, aber ich weiß nicht wie man es berechnet.Gibt es irgendwelche Werkzeuge, es zu tun,bitte?
- Duplikat von: stackoverflow.com/q/41996593/1714410
- das macht die Frage nicht beantworten. Die Auflösung dieser Verknüpfung ist, dass die Hälfte des Problems ist eine offene Anfrage in TF. Das ist Caffe.
Du musst angemeldet sein, um einen Kommentar abzugeben.
Zum online-tool finden Sie unter http://dgschwend.github.io/netscope/#/editor . Für alexnet sehen http://dgschwend.github.io/netscope/#/preset/alexnet . Dies unterstützt die meisten breiten Schichten bekannt. Für benutzerdefinierte Ebenen, die Sie haben, zu berechnen, sich.
Für zukünftige Besucher, wenn Sie Keras und TensorFlow als Backend, dann können Sie versuchen Sie das folgende Beispiel. Es berechnet die FLOPs für die MobileNet.
Wenn Sie mit Keras, du könntest einfach den patch in diesem pull-request: https://github.com/fchollet/keras/pull/6203
Dann rufen Sie print_summary() und Sie werden sehen, sowohl die flops pro Schicht und insgesamt.
Wenn auch nicht mit Keras, kann es sich lohnen, zu erstellen Sie Ihre Netze in Keras nur so können Sie den flops zählt.