Spaces:
Sleeping
Sleeping
File size: 255 Bytes
98a3af2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
"""
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)
|