Pierre Tassel commited on
Commit
8da99eb
·
1 Parent(s): d746b98

fix ray issue

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import time
5
  import multiprocessing as mp
6
  import json
7
 
 
8
  from PIL import Image
9
  from compiled_jss.CPEnv import CompiledJssEnvCP
10
 
@@ -46,6 +47,14 @@ def make_env(seed, instance):
46
 
47
 
48
  def solve(file):
 
 
 
 
 
 
 
 
49
  random.seed(0)
50
  np.random.seed(0)
51
  torch.manual_seed(0)
 
5
  import multiprocessing as mp
6
  import json
7
 
8
+ import ray
9
  from PIL import Image
10
  from compiled_jss.CPEnv import CompiledJssEnvCP
11
 
 
47
 
48
 
49
  def solve(file):
50
+ ray.init(log_to_driver=False,
51
+ reuse_actors=True,
52
+ include_dashboard=False, runtime_env={
53
+ "env_vars": {
54
+ # force `ray` to not kill a proces on OOM but use SWAP instead
55
+ "RAY_DISABLE_MEMORY_MONITOR": "1",
56
+ }
57
+ })
58
  random.seed(0)
59
  np.random.seed(0)
60
  torch.manual_seed(0)