Skip to content

Commit 61fb248

Browse files
committed
use gpu
1 parent 3b50f5e commit 61fb248

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/rank/naml/config_bigdata.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
runner:
1616
train_data_dir: "../../../datasets/MIND/data/train"
1717
train_reader_path: "NAMLDataReader" # importlib format
18-
use_gpu: False
18+
use_gpu: True
1919
train_batch_size: 50
2020
epochs: 2
2121
print_interval: 10

0 commit comments

Comments
 (0)