Research Article

Pipelined Training with Stale Weights in Deep Convolutional Neural Networks

Table 7

Memory usage of 4-stage pipelined ResNet training.

CNNPPVActivationsWeights (MB)IncreaseIncrease percentage in upper bound

ResNet-56(19)10.87 MB × batch size3.256.32 MB × batch size58
ResNet-110(37)21.43 MB × batch size6.5912.35 MB × batch size57
ResNet-224(75)43.70 MB × batch size13.6425.07 MB × batch size57
ResNet-362(121)70.67 MB × batch size22.1740.50 MB × batch size57