Save positional_embedding_plot figure using save_fig()

main
Aurélien Geron 2021-03-11 15:11:44 +13:00
parent b46db0e1a1
commit 2f2f9c8dd3
1 changed files with 1 additions and 1 deletions

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()"
]
},