|
@@ -122,7 +122,7 @@ class Main(object):
|
|
|
is_collective=False,
|
|
|
init_gloo=False
|
|
|
)
|
|
|
- fleet.init(role,config=fleet_config)
|
|
|
+ fleet.init(role)
|
|
|
|
|
|
|
|
|
|
|
@@ -131,9 +131,8 @@ class Main(object):
|
|
|
|
|
|
else:
|
|
|
|
|
|
-
|
|
|
-
|
|
|
- fleet.init(config=fleet_config)
|
|
|
+ fleet.init()
|
|
|
+ fleet.set_fleet_desc(fleet_config)
|
|
|
|
|
|
def network(self):
|
|
|
self.model = get_model(self.config)
|