[Fix] Call clip gradients if clip val greater than 0 (#6330)

* Call clip gradients if clip val greater than 0 * format * Format * Move to top of file
Lightning-AI · Mar 9, 2021 · 4abf247 · 4abf247
1 parent d8f90bc
commit 4abf247
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/plugins/test_sharded_plugin.py b/tests/plugins/test_sharded_plugin.py
@@ -31,6 +31,10 @@ def test_ddp_sharded_precision_16_clip_gradients(mock_oss_clip_grad_norm, clip_v
         mock_oss_clip_grad_norm.assert_not_called()
 
 
+<<<<<<< HEAD
+=======
+@RunIf(fairscale=True)
+>>>>>>> d0e7ca36... [Fix] Call clip gradients if clip val greater than 0 (#6330)
 @pytest.mark.parametrize(["accelerator"], [("ddp_sharded", ), ("ddp_sharded_spawn", )])
 @pytest.mark.skipif(not _FAIRSCALE_AVAILABLE, reason="Fairscale is not available")
 def test_sharded_ddp_choice(tmpdir, accelerator):