ddp pickle

2020-04-27 07:21:58 -04:00 · 2020-04-27 07:21:58 -04:00 · b993a3ed39
parent cebc74f7bc
commit b993a3ed39
1 changed files with 5 additions and 0 deletions
--- a/tests/trainer/test_trainer_cli.py
+++ b/tests/trainer/test_trainer_cli.py
@ -47,6 +47,11 @@ def test_add_argparse_args_redefined(cli_args):
        assert depr_name not in args

    trainer = Trainer.from_argparse_args(args=args)
+
+    # make sure trainer can be pickled
+    import pickle
+    pickle.dumps(trainer)
+
    assert isinstance(trainer, Trainer)