|
@@ -113,13 +113,16 @@ class Main(object):
|
|
|
logger.info("Run Success, Exit.")
|
|
|
|
|
|
def init_fleet_with_gloo(use_gloo=True):
|
|
|
+ fleet_config = {
|
|
|
+ "max_body_size": 256 * 1024 * 1024,
|
|
|
+ }
|
|
|
if use_gloo:
|
|
|
os.environ["PADDLE_WITH_GLOO"] = "0"
|
|
|
role = role_maker.PaddleCloudRoleMaker(
|
|
|
is_collective=False,
|
|
|
init_gloo=False
|
|
|
)
|
|
|
- fleet.init(role)
|
|
|
+ fleet.init(role,config=fleet_config)
|
|
|
|
|
|
|
|
|
|
|
@@ -127,7 +130,10 @@ class Main(object):
|
|
|
|
|
|
|
|
|
else:
|
|
|
- fleet.init()
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ fleet.init(config=fleet_config)
|
|
|
|
|
|
def network(self):
|
|
|
self.model = get_model(self.config)
|