Spaces:
Running
on
Zero
Running
on
Zero
File size: 265 Bytes
e85fecb |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
"""
Copied from RT-DETR (https://github.com/lyuwenyu/RT-DETR)
Copyright(c) 2023 lyuwenyu. All Rights Reserved.
"""
import torch.cuda.amp as amp
from ..core import register
__all__ = ["GradScaler"]
GradScaler = register()(amp.grad_scaler.GradScaler)
|