copy.py

import os
import shutil
from pathlib import Path
import re

base_dir = Path("/data/xyh/DFCs/DFC/models")
dest_dir = Path("/data/xyh/hf-upload/spflut+ours")

for variant_dir in base_dir.iterdir():
    if not variant_dir.is_dir():
        continue

    # Find latest .pth file based on number in filename
    pth_files = list(variant_dir.glob("Model_*.pth"))
    if not pth_files:
        continue

    def extract_number(f):
        match = re.search(r"Model_(\d+)\\.pth", f.name)
        return int(match.group(1)) if match else -1

    latest_pth = max(pth_files, key=extract_number)

    # Find all logs
    log_files = list(variant_dir.glob("train*.log")) + list(variant_dir.glob("lutft*.log"))

    # LUT files
    lut_files = list(variant_dir.glob("*.npy"))

    # Create destination dir
    dest_variant = dest_dir / variant_dir.name
    dest_variant.mkdir(parents=True, exist_ok=True)

    # Copy files
    for file in [latest_pth] + log_files + lut_files:
        shutil.copy2(file, dest_variant / file.name)