torch.autocast

Software / App

PyTorch autocasting context manager for mixed precision (bfloat16) training to reduce memory and increase speed.

Mentioned in 1 video

Videos Mentioning torch.autocast

Andrej Karpathy

PyTorch autocasting context manager for mixed precision (bfloat16) training to reduce memory and increase speed.