rabukasim / tools /_legacy_scripts /ai /debug /debug_worker.py
trioskosmos's picture
Upload folder using huggingface_hub
463f868 verified
import multiprocessing
import os
import sys
# Ensure project root is in path
sys.path.append(os.getcwd())
from ai.train_optimized import create_env
def worker(rank):
print(f"Worker {rank} starting pid={os.getpid()}")
try:
env = create_env(rank)
print(f"Worker {rank} env created: {env}")
env.close()
except Exception as e:
import traceback
traceback.print_exc()
print(f"Worker {rank} failed: {e}")
if __name__ == "__main__":
multiprocessing.freeze_support()
print(f"Main process pid={os.getpid()}")
p = multiprocessing.Process(target=worker, args=(0,))
p.start()
p.join()
print(f"Main process done. Exitcode: {p.exitcode}")