PyTorch - Neural Networks to Functional Blocks

Training a deep learning algorithm involves the following steps −

Building a data pipepne

Building a network architecture

Evaluating the architecture using a loss function

Optimizing the network architecture weights using an optimization algorithm

Training a specific deep learning algorithm is the exact requirement of converting a neural network to functional blocks as shown below −

With respect to the above diagram, any deep learning algorithm involves getting the input data, building the respective architecture which includes a bunch of layers embedded in them.

If you observe the above diagram, the accuracy is evaluated using a loss function with respect to optimization of the weights of neural network.