Update ensemble/load.py

ensemble/load.py

@@ -27,8 +27,8 @@ class EnsembleModel(torch.nn.Module):
         self.model6 = model6
         self.models = [model1, model2, model3, model4, model5, model6]
         self.mode = mode
-        if mode not in ["min", "mean", "max"]:
-            raise ValueError("Mode must be 'min', 'mean', or 'max'.")
+        if mode not in ["min", "mean", "max", "none"]:
+            raise ValueError("Mode must be 'none', 'min', 'mean', or 'max'.")
         
     def forward(self, x):
         outputs = []