chess-gpt-eval-contrastive/cleanup.py

import os
import re

directory = "../chess-mamba-vs-xformer/out/Xformer/50M/"

for filename in os.listdir(directory):
    if filename.endswith(".pt"):
        if filename == "ckpt.pt":
            continue
        
        match = re.search(r'ckpt_(\d+)(?:b|g)', filename)
        if match:
            number = int(match.group(1))
            if number > 33219200:
                file_path = os.path.join(directory, filename)
                os.remove(file_path)