diff --git a/distiller/data_loggers/logger.py b/distiller/data_loggers/logger.py index 89ac4730fc938e2a2da01677096f3d10ae33167c..9cb02312ac4788e1d363eacae8cb6ff1a9e29a29 100755 --- a/distiller/data_loggers/logger.py +++ b/distiller/data_loggers/logger.py @@ -138,7 +138,7 @@ class TensorBoardLogger(DataLogger): self.tblogger.scalar_summary('sparsity-2D/weights/' + name, sparsity_2D(param)*100, epoch) - self.tblogger.scalar_summary("sprasity/weights/total", 100*(1 - sparse_params_size/params_size), epoch) + self.tblogger.scalar_summary("sparsity/weights/total", 100*(1 - sparse_params_size/params_size), epoch) self.tblogger.sync_to_file() def log_weights_filter_magnitude(self, model, epoch, multi_graphs=False): diff --git a/jupyter/compare_executions.ipynb b/jupyter/compare_executions.ipynb old mode 100644 new mode 100755 index bfe8f8e72c721aaad40835785aa4bdb21521a38d..f680c459317a988236f071c4b6f748574897cd4f --- a/jupyter/compare_executions.ipynb +++ b/jupyter/compare_executions.ipynb @@ -53,7 +53,7 @@ "\n", "# Choose which performance indicators you wish to graph\n", "tags = ['Peformance/Validation/Top1', 'Peformance/Validation/Loss', \n", - " 'sprasity/weights/total', 'Peformance/Training/Reg Loss']\n", + " 'sparsity/weights/total', 'Peformance/Training/Reg Loss']\n", "\n", "f, axs = plt.subplots(2, 2, figsize=(20,20))\n", "f.suptitle('Performance')\n",