/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
torch
/
cpu
/
amp
/
/usr/local/lib64/python3.6/site-packages/torch/cpu/amp
mkdir
upload
Name
Size
Mode
Actions
__pycache__/
-
0755
rm
autocast_mode.py
373
0644
edit
dl
rm
__init__.py
36
0644
edit
dl
rm
Edit:
/usr/local/lib64/python3.6/site-packages/torch/cpu/amp/autocast_mode.py
(373B)
import torch class autocast(torch.autocast_mode.autocast): r""" See :class:`torch.autocast`. ``torch.cpu.amp.autocast(args...)`` is equivalent to ``torch.autocast("cpu", args...)`` """ def __init__(self, enabled=True, dtype=torch.bfloat16, cache_enabled=True): super().__init__("cpu", enabled=enabled, dtype=dtype, cache_enabled=cache_enabled)
Save
cmd:
run