Merge pull request #707 from yinhaofeng/use_xpu

frankwhzhang · web-flow · commit eff6da2d9ab8 · 2022-02-10T14:22:16.000+08:00
use_xpu
diff --git a/doc/source/paddlerec/yaml.md b/doc/source/paddlerec/yaml.md
@@ -17,6 +17,7 @@
 |             infer_start_epoch            |    int    |                           >= 0                           |    是    |    初始化模型时从第几个epoch保留的参数开始加载（从0开始计数，包括本次）    |
 |             infer_end_epoch            |    int    |                           >= 0                           |    是    |    初始化模型时到第几个epoch保留的参数停止加载（从0开始技术，不包括本次）    |
 |             use_gpu            |    bool    |                           True/False                           |    是    |                       指定是否使用gpu，若为False则默认使用cpu                        |
+|             use_xpu            |    bool    |                           True/False                           |    是    |                       指定是否使用xpu，若为False则默认使用cpu                        |
 |             epochs            |    int    |                           >= 1                           |    是    |                       指定train阶段需要训练几个epoch                        |
 |             print_interval            |    int    |                           >= 1                           |    是    |                       训练指标打印batch间隔                        |
 |             use_auc            |    bool    |                           True/False                           |    否    |                       在每个epoch开始时重置auc指标的值                        |
diff --git a/doc/yaml.md b/doc/yaml.md
@@ -17,6 +17,7 @@
 |             infer_start_epoch            |    int    |                           >= 0                           |    是    |    初始化模型时从第几个epoch保留的参数开始加载（从0开始计数，包括本次）    |
 |             infer_end_epoch            |    int    |                           >= 0                           |    是    |    初始化模型时到第几个epoch保留的参数停止加载（从0开始技术，不包括本次）    |
 |             use_gpu            |    bool    |                           True/False                           |    是    |                       指定是否使用gpu，若为False则默认使用cpu                        |
+|             use_xpu            |    bool    |                           True/False                           |    是    |                       指定是否使用xpu，若为False则默认使用cpu                        |
 |             epochs            |    int    |                           >= 1                           |    是    |                       指定train阶段需要训练几个epoch                        |
 |             print_interval            |    int    |                           >= 1                           |    是    |                       训练指标打印batch间隔                        |
 |             use_auc            |    bool    |                           True/False                           |    否    |                       在每个epoch开始时重置auc指标的值                        |
diff --git a/models/rank/deepfm/config.yaml b/models/rank/deepfm/config.yaml
@@ -17,6 +17,7 @@ runner:
   train_data_dir: "data/sample_data/train"
   train_reader_path: "criteo_reader" # importlib format
   use_gpu: False
+  use_xpu: False   # Enable this option only if you have an xpu device
   use_auc: True
   train_batch_size: 2
   epochs: 3
diff --git a/models/rank/deepfm/config_bigdata.yaml b/models/rank/deepfm/config_bigdata.yaml
@@ -18,6 +18,7 @@ runner:
   train_data_dir: "../../../datasets/criteo/slot_train_data_full"
   train_reader_path: "criteo_reader" # importlib format
   use_gpu: True
+  use_xpu: False   # Enable this option only if you have an xpu device
   use_auc: False
   train_batch_size: 512
   epochs: 1
diff --git a/models/rank/wide_deep/config.yaml b/models/rank/wide_deep/config.yaml
@@ -18,6 +18,7 @@ runner:
   train_data_dir: "data/sample_data/train"
   train_reader_path: "criteo_reader" # importlib format
   use_gpu: False
+  use_xpu: False   # Enable this option only if you have an xpu device
   use_auc: True
   train_batch_size: 50
   epochs: 4
diff --git a/models/rank/wide_deep/config_bigdata.yaml b/models/rank/wide_deep/config_bigdata.yaml
@@ -19,6 +19,7 @@ runner:
   train_data_dir: "../../../datasets/criteo/slot_train_data_full"
   train_reader_path: "criteo_reader" # importlib format
   use_gpu: True
+  use_xpu: False   # Enable this option only if you have an xpu device
   use_auc: True
   train_batch_size: 512
   epochs: 4
