FM / amp

/usr/local/lib64/python3.6/site-packages/torch/cuda/amp

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
autocast_mode.py	4562	0644	edit dl rm
common.py	157	0644	edit dl rm
grad_scaler.py	25315	0644	edit dl rm
__init__.py	124	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/cuda/amp/__init__.py (124B)

from .autocast_mode import autocast, custom_fwd, custom_bwd # noqa: F401 from .grad_scaler import GradScaler # noqa: F401