Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Neural MMO
neural-mmo-starter-kit
Commits
5855f8f3
Commit
5855f8f3
authored
Sep 21, 2021
by
Siddhartha Laghuvarapu
Browse files
Fix rllib_wrapper import in Forge.py
parent
f43da181
Changes
1
Hide whitespace changes
Inline
Side-by-side
neuralmmo/Forge.py
View file @
5855f8f3
...
...
@@ -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
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment