Skip to content

Commit b0c630f

Browse files
committed
No need to seed anymore
1 parent 08a5c4b commit b0c630f

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/callbacks/test_pruning.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from torch import nn
2323
from torch.nn import Sequential
2424

25-
from pytorch_lightning import seed_everything, Trainer
25+
from pytorch_lightning import Trainer
2626
from pytorch_lightning.callbacks import ModelCheckpoint, ModelPruning
2727
from pytorch_lightning.utilities.exceptions import MisconfigurationException
2828
from tests.helpers import BoringModel
@@ -225,7 +225,6 @@ def apply_lottery_ticket_hypothesis(self):
225225

226226
@pytest.mark.parametrize("make_pruning_permanent", (False, True))
227227
def test_multiple_pruning_callbacks(tmpdir, caplog, make_pruning_permanent: bool):
228-
seed_everything(0)
229228
model = TestModel()
230229
pruning_kwargs = {
231230
'parameters_to_prune': [(model.layer.mlp_1, "weight"), (model.layer.mlp_3, "weight")],
@@ -281,7 +280,6 @@ def test_permanent_when_model_is_saved_multiple_times(tmpdir, caplog, on_train_e
281280
make sure a copy is pruned and not the trained model if we want to continue
282281
with the same pruning buffers.
283282
"""
284-
seed_everything(0)
285283

286284
class TestPruning(ModelPruning):
287285

0 commit comments

Comments
 (0)