Skip to content

Commit 02de39b

Browse files
authored
bugfix
options backprop and add_prev were broken
1 parent 8632eb7 commit 02de39b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

colabdesign/af/design.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ def _recycle(self, model_params, backprop=True):
151151
if mode in ["backprop","add_prev"]:
152152

153153
# recycles compiled into model, only need single-pass
154-
num_recycles = self.opt["num_recycles"] = self._runner.config.model.num_recycle
154+
num_recycles = self.opt["num_recycles"] = self._cfg.model.num_recycle
155155
aux = self._single(model_params, backprop)
156156

157157
else:
@@ -457,4 +457,4 @@ def template_predesign(self, iters=100, soft=True, hard=False,
457457
self.set_opt(hard=hard, soft=soft, dropout=dropout, temp=temp)
458458
for i in range(iters):
459459
self.opt["template"]["dropout"] = (i+1)/iters
460-
self.step(**kwargs)
460+
self.step(**kwargs)

0 commit comments

Comments
 (0)