Replaces ddp .spawn with subprocess (#2029)

* replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * replace ddp spawn with subprocess * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix * hot fix
2026-09-09 11:32:07 +08:00 · 2020-06-01 11:00:32 -04:00
parent fd38f52e55
commit 82a20296e3
19 changed files with 283 additions and 174 deletions
@@ -220,7 +220,7 @@ def test_early_stopping_no_val_step(tmpdir):
        default_root_dir=tmpdir,
        early_stop_callback=stopping,
        overfit_pct=0.20,
-        max_epochs=5,
+        max_epochs=2,
    )
    result = trainer.fit(model)

@@ -254,7 +254,7 @@ def test_model_checkpoint_with_non_string_input(tmpdir, save_top_k):
    trainer = Trainer(default_root_dir=tmpdir,
                      checkpoint_callback=checkpoint,
                      overfit_pct=0.20,
-                      max_epochs=5
+                      max_epochs=2
                      )
    trainer.fit(model)

@@ -275,7 +275,7 @@ def test_model_checkpoint_path(tmpdir, logger_version, expected):
    trainer = Trainer(
        default_root_dir=tmpdir,
        overfit_pct=0.2,
-        max_epochs=5,
+        max_epochs=2,
        logger=logger
    )
    trainer.fit(model)
@@ -16,7 +16,7 @@ def test_lr_logger_single_lr(tmpdir):
    lr_logger = LearningRateLogger()
    trainer = Trainer(
        default_root_dir=tmpdir,
-        max_epochs=5,
+        max_epochs=2,
        val_percent_check=0.1,
        train_percent_check=0.5,
        callbacks=[lr_logger]
@@ -39,7 +39,7 @@ def test_lr_logger_no_lr(tmpdir):
    lr_logger = LearningRateLogger()
    trainer = Trainer(
        default_root_dir=tmpdir,
-        max_epochs=5,
+        max_epochs=2,
        val_percent_check=0.1,
        train_percent_check=0.5,
        callbacks=[lr_logger]
@@ -60,7 +60,7 @@ def test_lr_logger_multi_lrs(tmpdir):
    lr_logger = LearningRateLogger()
    trainer = Trainer(
        default_root_dir=tmpdir,
-        max_epochs=10,
+        max_epochs=2,
        val_percent_check=0.1,
        train_percent_check=0.5,
        callbacks=[lr_logger]
@@ -87,7 +87,7 @@ def test_lr_logger_param_groups(tmpdir):
    lr_logger = LearningRateLogger()
    trainer = Trainer(
        default_root_dir=tmpdir,
-        max_epochs=5,
+        max_epochs=2,
        val_percent_check=0.1,
        train_percent_check=0.5,
        callbacks=[lr_logger]