Add run on cpu-only compatibility

This commit is contained in:
Simon Alibert
2024-03-03 12:47:26 +01:00
parent 661bda45ea
commit b33ec5a630
5 changed files with 100 additions and 95 deletions

View File

@@ -10,8 +10,7 @@ hydra:
name: default
seed: 1337
device: cuda
buffer_device: cuda
device: cuda # cpu
prefetch: 4
eval_freq: ???
save_freq: ???