pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Michael Carilli a726827ec8 Formatting changes for gradient scaling (#33832 )

Summary:
hard to get right locally...I can build the docs but never quite match what it looks like live.  the bullet point indentation was just an oversight.

Removing `Returns:` formatting tabs because they take up a lot of space when rendered and add no clarity.  Some functions in Pytorch [do use them](https://pytorch.org/docs/master/torch.html#torch.eye), but [many don't bother](https://pytorch.org/docs/master/torch.html#torch.is_tensor), so apparently some people shared my feelings (Not using them is in line with existing practice).
Pull Request resolved: https://github.com/pytorch/pytorch/pull/33832

Differential Revision: D20135581

Pulled By: ngimel

fbshipit-source-id: bc788a7e57b142f95c4fa5baf3fe01f94c45abd8

2020-02-28 11:40:48 -08:00

amp

Formatting changes for gradient scaling (#33832 )

2020-02-28 11:40:48 -08:00

__init__.py

[WIP] Reanimate gradient scaling API with original scale update heuristic (#33366 )

2020-02-25 19:00:34 -08:00

__init__.pyi

Add type hint for cuda.set_rng_state (#26200 )

2019-09-14 19:29:42 -07:00

_utils.py

Fix trivial typos in torch.cuda._utils (#16026 )