from google.colab import files
files.download("./OUTPUT_MODEL/G_latest.pth")
files.download("./OUTPUT_MODEL/config.json")