Skip to content

Commit

Permalink
use gpu: True as default for model training
Browse files Browse the repository at this point in the history
  • Loading branch information
svandenhaute committed Dec 22, 2024
1 parent d4cc067 commit cadc2e9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion psiflow/execution.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()):
Expand Down

0 comments on commit cadc2e9

Please sign in to comment.