suryadev1 commited on
Commit
bcf042f
·
verified ·
1 Parent(s): 6a4781d

changed cuda to true

Browse files
Files changed (1) hide show
  1. new_test_saved_finetuned_model.py +1 -1
new_test_saved_finetuned_model.py CHANGED
@@ -558,7 +558,7 @@ def train():
558
  parser.add_argument("-w", "--num_workers", type=int, default=7, help="dataloader worker size")
559
 
560
  # Later run with cuda
561
- parser.add_argument("--with_cuda", type=bool, default=False, help="training with CUDA: true, or false")
562
  parser.add_argument("--log_freq", type=int, default=10, help="printing loss every n iter: setting n")
563
  # parser.add_argument("--corpus_lines", type=int, default=None, help="total number of lines in corpus")
564
  parser.add_argument("--cuda_devices", type=int, nargs='+', default=None, help="CUDA device ids")
 
558
  parser.add_argument("-w", "--num_workers", type=int, default=7, help="dataloader worker size")
559
 
560
  # Later run with cuda
561
+ parser.add_argument("--with_cuda", type=bool, default=True, help="training with CUDA: true, or false")
562
  parser.add_argument("--log_freq", type=int, default=10, help="printing loss every n iter: setting n")
563
  # parser.add_argument("--corpus_lines", type=int, default=None, help="total number of lines in corpus")
564
  parser.add_argument("--cuda_devices", type=int, nargs='+', default=None, help="CUDA device ids")