addressed jareds comments

zTaoplus · Jan 5, 2021 · 7417078 · 7417078
1 parent 203235f
commit 7417078
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 17 deletions.
diff --git a/megatron/fp16_deprecated/__init__.py b/megatron/fp16_deprecated/__init__.py
diff --git a/megatron/fp16_deprecated/loss_scaler.py b/megatron/fp16_deprecated/loss_scaler.py
@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import torch
+"""For backward compatibility, we need the class definitions to deserialize."""
 
 class LossScaler:
  def __init__(self, scale=1):