fixed a bug

This commit is contained in:
2026-01-31 16:31:50 +08:00
parent 7511bb30a0
commit cd40f843db

View File

@@ -166,6 +166,13 @@ class VideoCleaner:
if __name__ == "__main__": if __name__ == "__main__":
import argparse import argparse
# 设置多进程启动方法为 'spawn',这在 CUDA 环境下更安全,能避免显存冲突
import multiprocessing
try:
multiprocessing.set_start_method('spawn', force=True)
except RuntimeError:
pass
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("dir", help="Target directory") parser.add_argument("dir", help="Target directory")
parser.add_argument("--days", type=int, default=30) parser.add_argument("--days", type=int, default=30)