Add recommender system demo fault tolerant mode #284
Conversation
helinwang
changed the title
demo/recommender_system/train_ft.py
Outdated
|
|
||
|
|
||
| def main(): | ||
| paddle.init(use_gpu=False) |
There was a problem hiding this comment.
Maybe we do not speficy use_gpu, cloud job will set this parameter according to paddlecloud submit -gpu
There was a problem hiding this comment.
Thanks! Curious how we set it? From here I have not see the code set the gpu flag: https://github.com/PaddlePaddle/Paddle/blob/develop/python/paddle/v2/__init__.py#L68
There was a problem hiding this comment.
All the environment variable startswith PADDLE_INIT will be converted to the parameterse of paddle.init, as https://github.com/PaddlePaddle/Paddle/blob/develop/python/paddle/v2/__init__.py#L71. And cloud servert will set this environment variable when -gpu greater 0, as https://github.com/PaddlePaddle/cloud/blob/develop/paddlecloud/paddlejob/paddle_job.py#L96
Depends on the framework change in #278