From 3ae3c1d95bfc33becbbd6fcf685d18677eb37849 Mon Sep 17 00:00:00 2001 From: ceci3 Date: Sun, 26 Sep 2021 20:17:52 +0800 Subject: [PATCH] fix set value in paddle 2.1.1 (#877) --- paddleslim/nas/ofa/layers.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/paddleslim/nas/ofa/layers.py b/paddleslim/nas/ofa/layers.py index 18ca0c15..98763623 100644 --- a/paddleslim/nas/ofa/layers.py +++ b/paddleslim/nas/ofa/layers.py @@ -981,10 +981,10 @@ class SuperBatchNorm2D(nn.BatchNorm2D): batch_norm_out = core.ops.batch_norm(input, weight, bias, mean, variance, mean_out_tmp, variance_out_tmp, *attrs) - self._mean[:feature_dim] = mean - self._variance[:feature_dim] = variance - mean_out[:feature_dim] = mean_out_tmp - variance_out[:feature_dim] = variance_out_tmp + self._mean[:feature_dim].set_value(mean) + self._variance[:feature_dim].set_value(variance) + mean_out[:feature_dim].set_value(mean_out_tmp) + variance_out[:feature_dim].set_value(variance_out_tmp) else: batch_norm_out = core.ops.batch_norm(input, weight, bias, self._mean, self._variance, @@ -1031,10 +1031,10 @@ class SuperSyncBatchNorm(nn.SyncBatchNorm): sync_batch_norm_out, _, _, _, _, _ = core.ops.sync_batch_norm( input, weight, bias, mean, variance, mean_out_tmp, variance_out_tmp, *attrs) - self._mean[:feature_dim] = mean - self._variance[:feature_dim] = variance - mean_out[:feature_dim] = mean_out_tmp - variance_out[:feature_dim] = variance_out_tmp + self._mean[:feature_dim].set_value(mean) + self._variance[:feature_dim].set_value(variance) + mean_out[:feature_dim].set_value(mean_out_tmp) + variance_out[:feature_dim].set_value(variance_out_tmp) else: sync_batch_norm_out, _, _, _, _, _ = core.ops.sync_batch_norm( input, weight, bias, self._mean, self._variance, mean_out, -- GitLab