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

[Tests, GPU, SLOW] fix a bunch of GPU hardcoded tests in Pytorch (#4468)

Browse Source 
* fix gpu slow tests in pytorch

* change model to device syntax
This commit is contained in:
Patrick von Platen
2020-05-19 21:35:04 +02:00
committed by GitHub
parent 5856999a9f
commit aa925a52fa
11 changed files with 28 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/contrib/run_transfo_xl.py
View File

@@ -80,7 +80,7 @@ def main():
# Load a pre-trained model
model = TransfoXLLMHeadModel.from_pretrained(args.model_name)
model = model.to(device)
model.to(device)
logger.info(
"Evaluating with bsz {} tgt_len {} ext_len {} mem_len {} clamp_len {}".format(