Analysis of the Application Efficiency of TensorFlow and PyTorch in Convolutional Neural Network

Sensors (Basel). 2022 Nov 16;22(22):8872. doi: 10.3390/s22228872.

Abstract

In this paper, we present an analysis of important aspects that arise during the development of neural network applications. Our aim is to determine if the choice of library can impact the system's overall performance, either during training or design, and to extract a set of criteria that could be used to highlight the advantages and disadvantages of each library under consideration. To do so, we first extracted the previously mentioned aspects by comparing two of the most popular neural network libraries-PyTorch and TensorFlow-and then we performed an analysis on the obtained results, with the intent of determining if our initial hypothesis was correct. In the end, the results of the analysis are gathered, and an overall picture of what tasks are better suited for what library is presented.

Keywords: PyTorch; TensorFlow; convolutional neural network; network design; network training.

MeSH terms

  • Neural Networks, Computer*

Grants and funding

This research received no external funding.