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

[examples] consistently use --gpus, instead of --n_gpu (#6315)

Browse Source
This commit is contained in:
Stas Bekman
2020-08-07 07:36:32 -07:00
committed by GitHub
parent 0e36e51515
commit 6695450a23
4 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/text-classification/README.md
View File

@@ -249,7 +249,7 @@ The results are the following:
Run `bash run_pl.sh` from the `glue` directory. This will also install `pytorch-lightning` and the requirements in `examples/requirements.txt`. It is a shell pipeline that will automatically download, pre-process the data and run the specified models. Logs are saved in `lightning_logs` directory.
Pass `--n_gpu` flag to change the number of GPUs. Default uses 1. At the end, the expected results are:
Pass `--gpus` flag to change the number of GPUs. Default uses 1. At the end, the expected results are:
```
TEST RESULTS {'val_loss': tensor(0.0707), 'precision': 0.852427800698191, 'recall': 0.869537067011978, 'f1': 0.8608974358974358}
@@ -294,7 +294,3 @@ Training with the previously defined hyper-parameters yields the following resul
```bash
acc = 0.7093812375249501
```