update versions and change worker threads for evaluation

brandontrabucco · Jan 25, 2021 · d3ac8dd · d3ac8dd
1 parent 148b2ab
commit d3ac8dd
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 10 deletions.
diff --git a/design_bench/__init__.py b/design_bench/__init__.py
@@ -70,7 +70,7 @@ def save_response(response, destination):
 register(
     'GFP-v1',
     'design_bench.tasks.gfp_v1:GFPV1Task',
-    kwargs=dict(split_percentile=100))
+    kwargs=dict(split_percentile=20))
 
 register(
     'Superconductor-v0',
@@ -141,9 +141,9 @@ def save_response(response, destination):
             x_file='ant_morphology_X.npy',
             y_file='ant_morphology_y.npy',
             split_percentile=20,
-            num_rollouts=16,
+            num_rollouts=1,
             rollout_horizon=100,
-            num_parallel=8))
+            num_parallel=1))
 
     register(
         'DKittyMorphology-v0',
@@ -158,9 +158,9 @@ def save_response(response, destination):
             x_file='dkitty_morphology_X.npy',
             y_file='dkitty_morphology_y.npy',
             split_percentile=40,
-            num_rollouts=16,
+            num_rollouts=1,
             rollout_horizon=100,
-            num_parallel=8))
+            num_parallel=1))
 
 except ImportError as e:
 

diff --git a/design_bench/tasks/gfp_v1.py b/design_bench/tasks/gfp_v1.py
@@ -19,7 +19,7 @@ def onehottify(x, n=None, dtype=float):
 class GFPV1Task(Task):
 
     def __init__(self,
-                 split_percentile=100,
+                 split_percentile=20,
                  internal_batch_size=1,
                  use_cuda=True):
         """Load the GFP data set which includes maps from discrete

diff --git a/design_bench/tasks/morphology_v0.py b/design_bench/tasks/morphology_v0.py
@@ -23,9 +23,9 @@ def __init__(self,
                  x_file='ant_morphology_X.npy',
                  y_file='ant_morphology_y.npy',
                  split_percentile=60,
-                 num_rollouts=16,
+                 num_rollouts=1,
                  rollout_horizon=100,
-                 num_parallel=8):
+                 num_parallel=1):
         """Load static datasets of weights and their corresponding
         expected returns from the disk
 

diff --git a/setup.py b/setup.py
@@ -31,15 +31,15 @@
 setup(
     name='design-bench',
     packages=find_packages(include=['design_bench', 'design_bench.*']),
-    version='1.2',
+    version='1.3',
     license='MIT',
     description='Design-Bench: Benchmarks for Data-Driven Offline Model-Based Optimization',
     long_description=LONG_DESCRIPTION,
     long_description_content_type='text/markdown',
     author='Brandon Trabucco',
     author_email='brandon@btrabucco.com',
     url='https://github.com/brandontrabucco/design-bench',
-    download_url='https://github.com/brandontrabucco/design-bench/archive/v1_2.tar.gz',
+    download_url='https://github.com/brandontrabucco/design-bench/archive/v1_3.tar.gz',
     keywords=['Offline', 'Benchmark', 'Model-Based Optimization'],
     install_requires=INSTALL_REQUIRES,
     extras_require={'all': EXTRA_REQUIRES},