We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c83684f commit 9354b3eCopy full SHA for 9354b3e
paddlenlp/transformers/model_utils.py
@@ -1360,8 +1360,8 @@ def recompute_disable(self):
1360
"""
1361
1362
def fn(layer):
1363
- if hasattr(layer, "enable_recompute") and (layer.enable_recompute is False or layer.enable_recompute == 0):
1364
- layer.enable_recompute = True
+ if hasattr(layer, "enable_recompute") and (layer.enable_recompute is True or layer.enable_recompute == 1):
+ layer.enable_recompute = False
1365
1366
self.apply(fn)
1367
0 commit comments