Add memory requirements to ray

parent 4a877e3b
......@@ -43,7 +43,10 @@ def createPolicies(config, mapPolicy):
def loadTrainer(config):
'''Create monolithic RLlib trainer object'''
torch.set_num_threads(1)
ray.init(local_mode=config.LOCAL_MODE)
ray.init(local_mode=config.LOCAL_MODE,
_memory = 4*1024*1024*1024,
object_store_memory = 200*1024*1024
)
#Register custom env
ray.tune.registry.register_env("Neural_MMO",
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment