Spaces:
No application file
No application file
File size: 253 Bytes
b26e93d |
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)
|