From b9bf4282b59145ed28a8ca3b6f3f08363786ed58 Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Wed, 9 May 2018 17:01:12 +0300
Subject: [PATCH] change the access to the data of 0-dimensional tensors
---
jupyter/agp_schedule.ipynb | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/jupyter/agp_schedule.ipynb b/jupyter/agp_schedule.ipynb
index cce0734..c565607 100644
--- a/jupyter/agp_schedule.ipynb
+++ b/jupyter/agp_schedule.ipynb
@@ -116,7 +116,7 @@
" \n",
" sparsity_levels.append(sparsity)\n",
" sparsity.backward()\n",
- " d_sparsity.append(current_epoch.grad.data[0])\n",
+ " d_sparsity.append(current_epoch.grad.item())\n",
" current_epoch.grad.data.zero_()\n",
" else:\n",
" sparsity_levels.append(sparsity)\n",
--
GitLab