AMR device 0

ALLAN-DIP · May 15, 2023 · 5daf1fc · 5daf1fc
1 parent fbdb40a
commit 5daf1fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fairdiplomacy_external/mila_api.py b/fairdiplomacy_external/mila_api.py
@@ -173,7 +173,7 @@ async def play_mila(
 
         num_beams   = 4
         batch_size  = 16
-        device = 'cuda:1'
+        device = 'cuda:0'
         model_dir  = '/diplomacy_cicero/fairdiplomacy/AMR/amrlib/amrlib/data/model_parse_xfm/checkpoint-9920/'
         self.inference = Inference(model_dir, batch_size=batch_size, num_beams=num_beams, device=device)