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