From 7fcd263f37773d2f1f3caedec59d99e434fe7a29 Mon Sep 17 00:00:00 2001 From: mariumof <99500633+mariumof@users.noreply.github.com> Date: Thu, 9 Jun 2022 14:48:41 -0700 Subject: [PATCH] Upped version AND supported PT version (#600) --- smdebug/_version.py | 2 +- smdebug/pytorch/utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/smdebug/_version.py b/smdebug/_version.py index 851009801..441304cd0 100644 --- a/smdebug/_version.py +++ b/smdebug/_version.py @@ -1 +1 @@ -__version__ = "1.0.17" +__version__ = "1.0.18" diff --git a/smdebug/pytorch/utils.py b/smdebug/pytorch/utils.py index 6dd3744c8..24a58d678 100644 --- a/smdebug/pytorch/utils.py +++ b/smdebug/pytorch/utils.py @@ -13,7 +13,7 @@ # Cached Pytorch Version PT_VERSION = version.parse(torch.__version__) -SUPPORTED_PT_VERSION_THRESHOLD = version.parse("1.12.0") +SUPPORTED_PT_VERSION_THRESHOLD = version.parse("1.13.0") def get_reduction_of_data(reduction_name, tensor_data, tensor_name, abs=False):