[colab] 将训练好的图片保存到google drive的同文件下

我用colab进行网络训练，现在图片已经得到了，但是我想把训练好的图片保存在py同目录下的文件中，我新建了文件夹“ae_images”，然后让生成的图片保存进去，却总是报错，希望可以指导一下，怎么把生成的图片保存进google drive中
部分代码如下

save_images(x_concat, 'ae_images/rec_epoch_%d.png'%epoch)

FileNotFoundError: [Errno 2] No such file or directory: 'ae_images/rec_epoch_0.png'

自己问，自己答吧。代码如下，前两行代码输入后会出现一个链接，点进去以后会让你重新登录drive账号，然后给你一串很长的code，复制下来，输入到程序结果给的提示框里，然后3-4行代码就是定为你drive具体的文目录下了，具体名字把句号替换掉就好。

from google.colab import drive
drive.mount('/content/drive')

import os
os.chdir('/content/drive/My Drive/。。。。。。。。')