Skip to content

Commit 4abf247

Browse files
SeanNarentchaton
authored andcommitted
[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
1 parent d8f90bc commit 4abf247

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/plugins/test_sharded_plugin.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,10 @@ def test_ddp_sharded_precision_16_clip_gradients(mock_oss_clip_grad_norm, clip_v
3131
mock_oss_clip_grad_norm.assert_not_called()
3232

3333

34+
<<<<<<< HEAD
35+
=======
36+
@RunIf(fairscale=True)
37+
>>>>>>> d0e7ca36... [Fix] Call clip gradients if clip val greater than 0 (#6330)
3438
@pytest.mark.parametrize(["accelerator"], [("ddp_sharded", ), ("ddp_sharded_spawn", )])
3539
@pytest.mark.skipif(not _FAIRSCALE_AVAILABLE, reason="Fairscale is not available")
3640
def test_sharded_ddp_choice(tmpdir, accelerator):

0 commit comments

Comments
 (0)