Fix typo in ddp_cifar10_example.py (#20)

Summary:
This PR fixes a typo: `se_distributed_checkpoint ` -> `use_distributed_checkpoint`.

Pull Request resolved: facebookresearch/optimizers#20

Reviewed By: shintaro-iwasaki

Differential Revision: D62909206

Pulled By: hjmshi

fbshipit-source-id: bf3813e2d25896345ad299f77fc5dd48f808dfad

distributed_shampoo/examples/ddp_cifar10_example.py

@@ -140,7 +140,7 @@
         raise ValueError(
             "Distributed checkpointing is only supported with DistributedShampoo!"
         )
-    if args.se_distributed_checkpoint and args.checkpoint_dir is None:
+    if args.use_distributed_checkpoint and args.checkpoint_dir is None:
         raise ValueError(
             "Trying to use distributed checkpointing but checkpoint directory is not provided!"
         )