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 bc7b750 commit 35385dfCopy full SHA for 35385df
bayesflow/__init__.py
@@ -32,14 +32,6 @@ def setup():
32
33
from bayesflow.utils import logging
34
35
- if keras.backend.backend() == "torch":
36
- # turn off gradients by default
37
- import torch
38
-
39
- torch.autograd.set_grad_enabled(False)
40
41
- logging.warning("Disabling gradients by default. Use\nwith torch.enable_grad():\nin custom training loops.")
42
43
logging.debug(f"Using backend {keras.backend.backend()!r}")
44
45
0 commit comments