Skip to content

Commit

Permalink
Fix params
Browse files Browse the repository at this point in the history
  • Loading branch information
SeanNaren committed Feb 25, 2021
1 parent 86d916c commit b783d78
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions tests/callbacks/test_early_stopping.py
Original file line number Diff line number Diff line change
Expand Up @@ -369,11 +369,17 @@ def on_train_end(self) -> None:
([EarlyStopping(monitor='abc'), EarlyStopping(monitor='cba', patience=3)], 3, None, 1),
([EarlyStopping(monitor='cba', patience=3),
EarlyStopping(monitor='abc')], 3, None, 1),
pytest.param(([EarlyStopping(monitor='abc'),
EarlyStopping(monitor='cba', patience=3)], 3, 'ddp_cpu', 2),
pytest.param([EarlyStopping(monitor='abc'),
EarlyStopping(monitor='cba', patience=3)],
3,
'ddp_cpu',
2,
marks=pytest.mark.skipif(sys.platform == "win32", reason="DDP not available on windows")),
pytest.param(([EarlyStopping(monitor='cba', patience=3),
EarlyStopping(monitor='abc')], 3, 'ddp_cpu', 2),
pytest.param([EarlyStopping(monitor='cba', patience=3),
EarlyStopping(monitor='abc')],
3,
'ddp_cpu',
2,
marks=pytest.mark.skipif(sys.platform == "win32", reason="DDP not available on windows")),
],
)
Expand Down

0 comments on commit b783d78

Please sign in to comment.