Method test2

tests/cdslib/script/test_base_train.py:380–383

Test cuda version.

(self)

        self._test(n_gpus=0)

    def test2(self):
        """Test cuda version."""
        if torch.cuda.is_available():
            self._test(n_gpus=1)

    def test3(self):
        """Test cuda distributed version."""