Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Test 02 branch #906

Merged
merged 19 commits into from
Apr 3, 2023
Merged

Test 02 branch #906

Changes from 1 commit
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
cb11150
yifei add dnn register part for uapi
CoderYiFei Mar 20, 2023
82f1dc5
yifei add register dnn part
CoderYiFei Mar 20, 2023
0b42ef9
Merge branch 'master' of https://github.com/CoderYiFei/PaddleRec into…
CoderYiFei Mar 21, 2023
eabdddd
yifei add rec runner.py
CoderYiFei Mar 24, 2023
977bcd0
[add function] yifei add rec runner.py
CoderYiFei Mar 24, 2023
7f53cc3
Merge branch 'master' into test_02_branch
CoderYiFei Mar 24, 2023
a09a5a3
yifei delete print funcs
CoderYiFei Mar 24, 2023
271b11d
Merge branch 'test_02_branch' of https://github.com/CoderYiFei/Paddle…
CoderYiFei Mar 24, 2023
a6c0934
yifei change gpu list length from 8 to 2
CoderYiFei Mar 29, 2023
dbafe35
Merge branch 'master' into test_02_branch
wangzhen38 Mar 29, 2023
e91d856
yifei add gpu list
CoderYiFei Mar 31, 2023
221b76e
Merge branch 'master' into test_02_branch
wangzhen38 Mar 31, 2023
46c4981
yifei change device name
CoderYiFei Mar 31, 2023
d87e251
Merge branch 'test_02_branch' of https://github.com/CoderYiFei/Paddle…
CoderYiFei Mar 31, 2023
330975e
yifei change device name
CoderYiFei Mar 31, 2023
1cd16cf
yifei change dataset part
CoderYiFei Apr 3, 2023
2069329
Merge branch 'master' into test_02_branch
wangzhen38 Apr 3, 2023
4b3284d
yifei change data check funcs
CoderYiFei Apr 3, 2023
1dd1d56
Merge branch 'test_02_branch' of https://github.com/CoderYiFei/Paddle…
CoderYiFei Apr 3, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
yifei add gpu list
  • Loading branch information
@CoderYiFei
CoderYiFei committed Mar 31, 2023
commit e91d8560c599ce5ae5ff92c783c12a94bd2cf959
7 changes: 7 additions & 0 deletions uapi_rec/rank/runner.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,11 @@ def train(self, config_path, cli_args, device, ips):
learning_rate_value = 0
iters_value = 0
save_dir_value = ""
device = "cpu"
try:
batch_size_value = args_str.split("--device ")[1].split(" ")[0]
except:
pass
try:
batch_size_value = args_str.split("--batch_size ")[1].split(" ")[0]
except:
Expand All @@ -49,6 +54,8 @@ def train(self, config_path, cli_args, device, ips):
paddle_pserver_ip_port = "127.0.0.1:29011"
#gpus_list = "0,1,2,3,4,5,6,7"
gpus_list = "0,1"
if "device" in device:
gpus_list = device.lstrip("gpu:")
server_echo = "PADDLE WILL START PSERVER 29011"
trainer_echo = "PADDLE WILL START Trainer 0"
cmd = f"export FLAGS_LAUNCH_BARRIER=0 && \
Expand Down