Skip to content
Snippets Groups Projects
Commit 074d6330 authored by Pat Alt's avatar Pat Alt
Browse files

added gpu accelaration

parent 40804726
No related branches found
No related tags found
1 merge request!7669 initial run including fmnist lenet and new method
Hide whitespace changes
Inline Side-by-side
experiments/models/default_models.jl
+ 3
− 1
View file @ 074d6330
...@@ -49,6 +49,7 @@ function default_models(; ...@@ -49,6 +49,7 @@ function default_models(;
batch_size=batch_size, batch_size=batch_size,
finaliser=finaliser, finaliser=finaliser,
loss=loss, loss=loss,
acceleration=CUDALibs(),
) )
# Deep Ensemble: # Deep Ensemble:
...@@ -65,7 +66,8 @@ function default_models(; ...@@ -65,7 +66,8 @@ function default_models(;
jem_training_params=( jem_training_params=(
α=α, verbosity=verbosity, α=α, verbosity=verbosity,
), ),
sampling_steps=sampling_steps sampling_steps=sampling_steps,
acceleration=CUDALibs(),
) )
# Deep Ensemble of Joint Energy Models: # Deep Ensemble of Joint Energy Models:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment