Method __init__

index/dense/libs.py:246–252

(self, activated, use_cuda)

class MixedPrecisionManager:
    def __init__(self, activated, use_cuda):
        assert (not activated) or PyTorch_over_1_6, "Cannot use AMP for PyTorch version < 1.6"

        print("Using FP16:", activated)
        self.activated = activated and use_cuda
        if self.activated:
            self.scaler = torch.cuda.amp.GradScaler()

    def context(self):
        return torch.cuda.amp.autocast() if self.activated else NullContextManager()