SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Harmonize no_cuda flag with other scripts

Browse Source
This commit is contained in:
Julien Chaumond
2019-12-10 01:32:42 +00:00
parent f7eba09007
commit 1d18930462
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/summarization/run_summarization.py
View File

@@ -272,7 +272,7 @@ def main():
) )
# EVALUATION options # EVALUATION options
parser.add_argument( parser.add_argument(
"--to_cpu", "--no_cuda",
default=False, default=False,
type=bool, type=bool,
help="Whether to force the execution on CPU.", help="Whether to force the execution on CPU.",
@@ -314,7 +314,7 @@ def main():
args = parser.parse_args() args = parser.parse_args()
# Select device (distibuted not available) # Select device (distibuted not available)
args.device = torch.device("cuda" if torch.cuda.is_available() and not args.to_cpu else "cpu") args.device = torch.device("cuda" if torch.cuda.is_available() and not args.no_cuda else "cpu")
# Check the existence of directories # Check the existence of directories
if not args.summaries_output_dir: if not args.summaries_output_dir:

2
requirements.txt
View File

@@ -9,4 +9,4 @@ regex
# For XLNet # For XLNet
sentencepiece sentencepiece
# For XLM # For XLM
sacremoses sacremoses