Spaces:
Running
Running
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")
|