bugfix

options backprop and add_prev were broken
sokrypton · Sep 5, 2022 · 02de39b · 02de39b
1 parent 8632eb7
commit 02de39b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/colabdesign/af/design.py b/colabdesign/af/design.py
@@ -151,7 +151,7 @@ def _recycle(self, model_params, backprop=True):
     if mode in ["backprop","add_prev"]:
 
       # recycles compiled into model, only need single-pass
-      num_recycles = self.opt["num_recycles"] = self._runner.config.model.num_recycle
+      num_recycles = self.opt["num_recycles"] = self._cfg.model.num_recycle
       aux = self._single(model_params, backprop)
 
     else:
@@ -457,4 +457,4 @@ def template_predesign(self, iters=100, soft=True, hard=False,
     self.set_opt(hard=hard, soft=soft, dropout=dropout, temp=temp)
     for i in range(iters):
       self.opt["template"]["dropout"] = (i+1)/iters
-      self.step(**kwargs)
+      self.step(**kwargs)