PyTorch Recipes: A Problem-Solution Approach to Build, Train and Deploy Neural Network Models

© The Author(s), under exclusive license to APress Media, LLC, part of Springer Nature 2023

P. MishraPyTorch Recipeshttps://doi.org/10.1007/978-1-4842-8925-9_3

3. CNN and RNN Using PyTorch

Pradeepta Mishra¹  

(1)

Bangalore, Karnataka, India

 

This chapter covers convolutional neural networks (CNN) and recurrent neural network and their implementation using PyTorch . Convolutional neural network is being used in image classification, object detection, object classification related tasks. The large scale image classification models requires PyTorch framework as it is considered to be faster than other frameworks, similarly the recurrent neural networks are used in natural language processing tasks such as text classification, sentiment classification, ...