Method test1

tests/cdslib/optim/test_tf_optim.py:72–75

(self)

                scaler.update()

    def test1(self):
        if torch.__version__ >= "1.7.0" and torch.cuda.is_available():
            # AMP is better supported only on cuda
            self._test(use_amp=True)

    def test2(self):
        self._test(use_amp=False)