From 6f7ea776b2234fbd7bde21b662acc1f7f36baf84 Mon Sep 17 00:00:00 2001 From: Aniketos07 <51787443+Aniketos07@users.noreply.github.com> Date: Thu, 27 Jan 2022 16:37:15 +0530 Subject: [PATCH] tf.Session() deprecated, changed to tf.compat.v1.Session() --- python/dpu_utils/tfutils/gradratiologgingoptimizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/dpu_utils/tfutils/gradratiologgingoptimizer.py b/python/dpu_utils/tfutils/gradratiologgingoptimizer.py index a613a8a..d18c288 100644 --- a/python/dpu_utils/tfutils/gradratiologgingoptimizer.py +++ b/python/dpu_utils/tfutils/gradratiologgingoptimizer.py @@ -16,7 +16,7 @@ def __init__(self, optimizer, name='training-optimizer'): def optimizer(self): return self.__optimizer - def print_ratios(self, session: tf.Session): + def print_ratios(self, session: tf.compat.v1.Session): count = self.__acc_count.eval(session) + 1e-10 print('======================') print('Gradient Ratios') @@ -39,4 +39,4 @@ def minimize(self, loss): update_ops.append(tf.assign_add(ratio_acc_var, grad_ratio)) grad_apply_op = self.__optimizer.apply_gradients(gradients_and_vars) update_ops.append(grad_apply_op) - return control_flow_ops.group(*update_ops) \ No newline at end of file + return control_flow_ops.group(*update_ops)