Skip to content

Latest commit

 

History

History
11 lines (7 loc) · 366 Bytes

README.md

File metadata and controls

11 lines (7 loc) · 366 Bytes

GPUUtilizationCallback

A fastai callback to plot GPU usage during training. It's based on the data taken from the nvidia-smi stats command.

Installation

None for now, copy the callback class into your code.

Usage

Pass the GPUUtilizationCallback into the cbs argument of the Learner. See the notebook for examples

example