MCPcopy
hub / github.com/PaddlePaddle/PaddleFormers / get_gpus

Function get_gpus

tests/transformers/test_modeling_common.py:73–75  ·  view source on GitHub ↗
(selected_gpus)

Source from the content-addressed store, hash-verified

71
72
73def get_gpus(selected_gpus):
74 selected_gpus = [x.strip() for x in selected_gpus.split(",")]
75 return selected_gpus
76
77
78def start_local_trainers_cpu(trainer_endpoints, training_script, training_script_args, log_dir=None):

Callers 1

run_on_gpuMethod · 0.70

Calls 1

splitMethod · 0.45

Tested by

no test coverage detected