fix

DingXiaoH · Jan 8, 2021 · a10ec1c · a10ec1c
1 parent 1762a4a
commit a10ec1c
Show file tree

Hide file tree

Showing 2 changed files with 490 additions and 1 deletion.
diff --git a/csgd/csgd_pipeline.py b/csgd/csgd_pipeline.py
@@ -1,7 +1,7 @@
 from ndp_train import train_main
 import os
 from ndp_test import general_test
-from csgd.flops_scripts import csgd_train_main
+from csgd.ddp_csgd_train import csgd_train_main
 
 def csgd_prune_pipeline(local_rank, init_hdf5, base_train_config, csgd_train_config,
                         target_deps, centri_strength, pacesetter_dict, succeeding_strategy):