From 074d63302337ce8ee9f8c79a6cecc626e9f7467b Mon Sep 17 00:00:00 2001
From: Pat Alt <55311242+pat-alt@users.noreply.github.com>
Date: Tue, 12 Sep 2023 13:06:18 +0200
Subject: [PATCH] added gpu accelaration

---
 experiments/models/default_models.jl | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/experiments/models/default_models.jl b/experiments/models/default_models.jl
index 53319cd9..13415159 100644
--- a/experiments/models/default_models.jl
+++ b/experiments/models/default_models.jl
@@ -49,6 +49,7 @@ function default_models(;
         batch_size=batch_size,
         finaliser=finaliser,
         loss=loss,
+        acceleration=CUDALibs(),
     )
 
     # Deep Ensemble:
@@ -65,7 +66,8 @@ function default_models(;
         jem_training_params=(
             α=α, verbosity=verbosity,
         ),
-        sampling_steps=sampling_steps
+        sampling_steps=sampling_steps,
+        acceleration=CUDALibs(),
     )
 
     # Deep Ensemble of Joint Energy Models:
-- 
GitLab