fix cluster

fleet_rec/core/trainers/cluster_trainer.py

@@ -103,7 +103,7 @@ class ClusterTrainer(TranspileTrainer):
         self._exe.run(fleet.startup_program)
         context['status'] = 'train_pass'
 
-   def dataloader_train(self, context):
+    def dataloader_train(self, context):
         fleet.init_worker()
 
         reader = self._get_dataloader()

fleet_rec/core/trainers/tdm_cluster_trainer.py

@@ -19,6 +19,7 @@ Training use fluid with one node only.
 
 from __future__ import print_function
 import logging
+import numpy as np
 import paddle.fluid as fluid
 from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler import fleet
 from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler.distributed_strategy import StrategyFactory