fix: fix baguaelastic launch script

BaguaSys · Jun 10, 2021 · b069cd4 · b069cd4
1 parent 56c9cee
commit b069cd4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/baguaelastic/distributed/launch.py b/baguaelastic/distributed/launch.py
@@ -29,6 +29,7 @@
 from torchelastic.agent.server.api import WorkerSpec, WorkerState
 from torchelastic.agent.server.local_elastic_agent import LocalElasticAgent
 from torchelastic.distributed.argparse_util import check_env, env
+import torch
 
 
 log = get_logger()