We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c700590 commit 9c13878Copy full SHA for 9c13878
tests/plugins/test_custom_plugin.py
@@ -11,6 +11,9 @@
11
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
# See the License for the specific language governing permissions and
13
# limitations under the License.
14
+import pytest
15
+import torch
16
+
17
from pytorch_lightning import Trainer
18
from pytorch_lightning.plugins import DDPPlugin
19
from tests.helpers import BoringModel
@@ -26,6 +29,7 @@ def __init__(self, **kwargs):
26
29
27
30
28
31
@RunIf(skip_windows=True)
32
+@pytest.mark.skipif(torch.cuda.is_available(), reason="RuntimeError: Tensors must be CUDA and dense")
33
def test_sync_batchnorm_set(tmpdir):
34
"""Tests if sync_batchnorm is automatically set for custom plugin."""
35
model = BoringModel()
0 commit comments