Save positional_embedding_plot figure using save_fig()

This commit is contained in:
Aurélien Geron
2021-03-11 15:11:44 +13:00
parent b46db0e1a1
commit 2f2f9c8dd3

View File

@@ -1140,7 +1140,7 @@
"ax2.axis([0, max_steps - 1, 0, crop_i])\n",
"ax2.set_xlabel(\"$p$\", fontsize=16)\n",
"ax2.set_ylabel(\"$i$\", rotation=0, fontsize=16)\n",
"plt.savefig(\"positional_embedding_plot\")\n",
"save_fig(\"positional_embedding_plot\")\n",
"plt.show()"
]
},