Spaces:
Sleeping
Sleeping
File size: 747 Bytes
463f868 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | 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}")
|