Update.

diff --git a/main.py b/main.py
index aefc3a1..ba5f04b 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -246,7 +246,6 @@ if args.problem == "sky":
     back_accuracy = False
 elif args.problem == "grids":
     problem = grids.Grids(
-        device=device,
         max_nb_cached_chunks=args.nb_train_samples // 100,
         chunk_size=100,
         nb_threads=args.nb_threads,