diff --git a/cifar/cifar10_deepspeed.py b/cifar/cifar10_deepspeed.py index ea5c7d8fe..64e488816 100755 --- a/cifar/cifar10_deepspeed.py +++ b/cifar/cifar10_deepspeed.py @@ -108,8 +108,6 @@ def forward(self, x): net = Net() parameters = filter(lambda p: p.requires_grad, net.parameters()) args=add_argument() -# Hack to disable client scheduler -args.use_scheduler = False # Initialize DeepSpeed to use the following features # 1) Distributed model