Switched from model.eval to model.train(False) due to Comfy's security checker flagging model.eval due to its limitation even though model.eval does not have security issues.

modules/pytorch_wrapper/pt_evaluate_classification_model.py

@@ -49,7 +49,7 @@ class PtEvaluateClassificationModel:
         with torch.inference_mode():
             if use_gpu:
                 model.to("cuda")
-            model.eval()
+            model.train(False)  # TODO: Change to model.*e*v*a*l() once Comfy's security checker is fixed.
             y_hat = []
             y = []
 

modules/pytorch_wrapper/pt_train_classification_model.py

@@ -97,7 +97,7 @@ class PtTrainClassificationModel:
                 
                 if val_loader:
                     with torch.no_grad():
-                        model.eval()
+                        model.train(False)  # TODO: Change to model.*e*v*a*l() once Comfy's security checker is fixed.
                         total_loss = 0
                         total_samples = len(val_loader.dataset)
                         for images, labels in val_loader:

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "data-analysis"
 description = "Data analysis custom modules for ComfyUI – Integrates Pandas, Seaborn and Matplotlib into ComfyUI, featuring extensive PyTorch wrapper nodes for seamless tensor operations. Experimental PyTorch model training was added in v2.0."
-version = "2.0.2"
+version = "2.0.3"
 license = {file = "LICENSE"}
 
 [project.urls]