Save positional_embedding_plot figure using save_fig()
parent
b46db0e1a1
commit
2f2f9c8dd3
|
@ -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()"
|
||||
]
|
||||
},
|
||||
|
|
Loading…
Reference in New Issue