arguments

Janetalready · Sep 21, 2022 · bf0a07c · bf0a07c
1 parent a403532
commit bf0a07c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils_dreamer.py b/utils_dreamer.py
@@ -33,6 +33,7 @@ def parse_arguments():
         '--resume_path', default=None, help='pretrained folder')
     parser.add_argument(
         '--setting', default='complete', help='game settings')
+    parser.add_argument('--cuda', type=int, default=1, help='enables cuda')
     parser.add_argument('--max_step', type=int,
                         help='number of drawing steps', default=10)
     parser.add_argument('--num_stroke', type=int,