# ガーベージコレクターのライブラリをインポート
import gc

# 不要なオブジェクトを解放し、メモリ最適化する。
gc.collect()

google colabなどメモリに限りがある環境を利用してる場合は、このコードを適所に入れることで最適化する。

おすすめの記事