No gradients before torch predictions

ahmedfgad · Sep 21, 2024 · e17b999 · e17b999
1 parent 9ffa344
commit e17b999
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pygad/torchga/__init__.py b/pygad/torchga/__init__.py
@@ -1,3 +1,3 @@
 from .torchga import *
 
-__version__ = "1.3.0"
+__version__ = "1.4.0"
diff --git a/pygad/torchga/torchga.py b/pygad/torchga/torchga.py
@@ -44,9 +44,10 @@ def predict(model, solution, data):
     _model = copy.deepcopy(model)
     _model.load_state_dict(model_weights_dict)
 
-    predictions = _model(data)
+    with torch.no_grad():
+        predictions = _model(data)
 
-    return predictions
+        return predictions
 
 class TorchGA: