Logo image
Optimization of GPU Memory Usage for Training Deep Neural Networks
Conference paper   Peer reviewed

Optimization of GPU Memory Usage for Training Deep Neural Networks

Che-Lun Hung, Chine-fu Hsin, Hsiao-Hsi Wang and Chuan Yi Tang
Communications in Computer and Information Science, Vol.1080 CCIS, pp.289-293
2019

Abstract

Convolutional Neural Networks Deep Neural Network GPU Computer Science (all) Mathematics (all)
Recently, Deep Neural Networks have been successfully utilized in many domains; especially in computer vision. Many famous convolutional neural networks, such as VGG, ResNet, Inception, and so forth, are used for image classification, object detection, and so forth. The architecture of these state-of-the-art neural networks has become deeper and complicated than ever. In this paper, we propose a method to solve the problem of large memory requirement in the process of training a model. The experimental result shows that the proposed algorithm is able to reduce the GPU memory significantly.

Metrics

1 Record Views

Details

Logo image