use gpu: True as default for model training

svandenhaute · svandenhaute · commit cadc2e9fdc9e · 2024-12-22T14:11:50.000+01:00
diff --git a/psiflow/execution.py b/psiflow/execution.py
@@ -488,7 +488,7 @@ def from_config(
         )
         model_training = ModelTraining.from_config(
             container=container,
-            **kwargs.pop("ModelTraining", {}),
+            **kwargs.pop("ModelTraining", {'gpu': True}),  # avoid triggering assertion
         )
         reference_evaluations = []  # reference evaluations might be class specific
         for key in list(kwargs.keys()):

Original file line number	Diff line number	Diff line change
`@@ -488,7 +488,7 @@ def from_config(`
`488`	`488`	`)`
`489`	`489`	`model_training = ModelTraining.from_config(`
`490`	`490`	`container=container,`
`491`		`- **kwargs.pop("ModelTraining", {}),`
	`491`	`+ **kwargs.pop("ModelTraining", {'gpu': True}), # avoid triggering assertion`
`492`	`492`	`)`
`493`	`493`	`reference_evaluations = [] # reference evaluations might be class specific`
`494`	`494`	`for key in list(kwargs.keys()):`