Comparaison Keras - Pytorch

Question

Une comparaison a-t-elle été effectuée entre les performances des modèles Keras du notebook (01-DNN-MNIST) et celles de Pytorch du notebook 01DNN-MNIST-Pytorch) en utilisant exactement les mêmes données de départ ?

Answer 1

Pour cet exemple précis non, mais beaucoup de benchmark pytorch vs tensorflow existe. Un résumé serait que pytorch serait toujours plus performant s'il est bien optimisé (sauf si vous avez des TPUs)

Comments

Alors cela ferait un exercice intéressant sur les données "MNIST-1" de Fidle.
Je vais essayer de récupérer le notebook 01-DNN-MNIST-Pytorch.

Answer 2

Sur ce notebook il est très court donc pas vraiment de comparaison entre les deux frameworks. Mais de manière générale PyTorch est bien plus rapide sur les gros réseaux et les applis complexes.

Comments

D'accord, merci.