import os CHECKPOINTS_DIR = os.path.dirname(os.path.abspath(__file__)) if __name__ == "__main__": print(CHECKPOINTS_DIR)