Spaces:
Sleeping
Sleeping
| 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}") | |