From dccdb3112e6c3d6e8b75b2ee20b270755c7a5751 Mon Sep 17 00:00:00 2001
From: Karthik Gopalakrishnan <karthgop@amazon.com>
Date: Tue, 1 Dec 2020 19:08:08 -0800
Subject: [PATCH] remove session_params key from docstrings

---
 deepspeed/runtime/activation_checkpointing/config.py | 2 --
 deepspeed/runtime/zero/constants.py                  | 2 --
 2 files changed, 4 deletions(-)

diff --git a/deepspeed/runtime/activation_checkpointing/config.py b/deepspeed/runtime/activation_checkpointing/config.py
index 30ac5157f843..b595a2a18bdf 100755
--- a/deepspeed/runtime/activation_checkpointing/config.py
+++ b/deepspeed/runtime/activation_checkpointing/config.py
@@ -12,7 +12,6 @@
 #activations for the backpropagation.
 ACTIVATION_CHKPT_FORMAT = '''
 Activation Checkpointing should be configured as:
-"session_params": {
   "activation_checkpointing": {
     "partitioned_activations": [true|false],
     "number_checkpoints": 100,
@@ -21,7 +20,6 @@
     "profile": [true|false],
     "synchronize_checkpoint_boundary": [true|false],
     }
-}
 '''
 
 ACT_CHKPT_PARTITION_ACTIVATIONS = 'partition_activations'
diff --git a/deepspeed/runtime/zero/constants.py b/deepspeed/runtime/zero/constants.py
index fd90033dc3f5..ba902b72f4c0 100755
--- a/deepspeed/runtime/zero/constants.py
+++ b/deepspeed/runtime/zero/constants.py
@@ -10,7 +10,6 @@
 # Users have to configure the desired optimization (0 means disabled) in params.json as below example:
 ZERO_FORMAT = '''
 ZeRO optimization should be enabled as:
-"session_params": {
   "zero_optimization": {
     "stage": [0|1|2],
     "allgather_partitions": [true|false],
@@ -22,7 +21,6 @@
     "load_from_fp32_weights": [true|false]
     "cpu_offload": [true|false]
     }
-}
 '''
 
 ZERO_OPTIMIZATION = 'zero_optimization'