Spaces:
Sleeping
Sleeping
changed cuda to true
Browse files
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=
|
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")
|