add gpu config

core/trainers/transpiler_trainer.py

@@ -28,6 +28,10 @@ from paddlerec.core.utils import dataloader_instance
 class TranspileTrainer(Trainer):
     def __init__(self, config=None):
         Trainer.__init__(self, config)
+        device = envs.get_global_env("train.device")
+        if device == 'gpu':
+            self._place = fluid.CUDAPlace(0)
+            self._exe = fluid.Executor(self._place)
         self.processor_register()
         self.model = None
         self.inference_models = []

models/multitask/esmm/config.yaml

@@ -19,6 +19,7 @@ train:
 
   epochs: 3
   workspace: "paddlerec.models.multitask.esmm"
+  device: cpu
 
   reader:
     batch_size: 2

models/multitask/mmoe/config.yaml

@@ -19,6 +19,7 @@ train:
 
   epochs: 3
   workspace: "paddlerec.models.multitask.mmoe"
+  device: cpu
 
   reader:
     batch_size: 2

models/multitask/share-bottom/config.yaml

@@ -19,6 +19,7 @@ train:
 
   epochs: 3
   workspace: "paddlerec.models.multitask.share-bottom"
+  device: cpu
 
   reader:
     batch_size: 2

models/recall/gru4rec/config.yaml

@@ -20,6 +20,7 @@ train:
   epochs: 3
   workspace: "paddlerec.models.recall.gru4rec"
 
+  device: cpu
   reader:
     batch_size: 5
     class: "{workspace}/rsc15_reader.py"

models/recall/ssr/config.yaml

@@ -19,6 +19,7 @@ train:
 
   epochs: 3
   workspace: "paddlerec.models.recall.ssr"
+  device: cpu
 
   reader:
     batch_size: 5