Commit 5855f8f3 authored by Siddhartha Laghuvarapu's avatar Siddhartha Laghuvarapu
Browse files

Fix rllib_wrapper import in Forge.py

parent f43da181
......@@ -16,7 +16,7 @@ import ray
from ray import rllib, tune
from ray.tune import CLIReporter
from ray.tune.integration.wandb import WandbLoggerCallback
from neuralmmo.projekt import rllib_wrapper as wrapper
from projekt import rllib_wrapper as wrapper
import projekt
from projekt import config as base_config
......@@ -178,6 +178,7 @@ class Anvil():
def loadTrainer(config):
'''Create monolithic RLlib trainer object'''
from neuralmmo.projekt import rllib_wrapper as wrapper
torch.set_num_threads(1)
ray.init(local_mode=config.LOCAL_MODE,
_memory=2000 * 1024 * 1024,
......
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