This repository provides Python scripts to monitor and calculate GPU memory consumption during PyTorch model training. It can estimate the memory footprint of individual model layers including parameters and intermediate variables for both forward and backward passes. The tool prints detailed memory breakdowns by tensor size, dtype, and class type, helping developers optimize memory usage when training deep learning models.