From ee97ec62641e731d922338519a36e8b06caf505c Mon Sep 17 00:00:00 2001 From: BigFishMaster Date: Fri, 23 Mar 2018 01:41:07 +0800 Subject: [PATCH] Add memory optimization amd nccl --- fluid/image_classification/se_resnext.py | 1 - 1 file changed, 1 deletion(-) diff --git a/fluid/image_classification/se_resnext.py b/fluid/image_classification/se_resnext.py index ab2eb668..c9f2f0cf 100644 --- a/fluid/image_classification/se_resnext.py +++ b/fluid/image_classification/se_resnext.py @@ -270,7 +270,6 @@ if __name__ == '__main__': lr_strategy = {"bd": bd, "lr": lr} use_nccl = True - # layers: 50, 152 layers = 50 -- GitLab