Commit f797d87a authored by Michał Pogoda's avatar Michał Pogoda

Automatically enable GPU in train script when gpu flag is provided

parent e23c94eb
Pipeline #2386 passed with stages
in 12 minutes and 26 seconds
......@@ -63,6 +63,7 @@ fi
if [ ! -z "$gpu" ]; then
docker_params+=(--gpus device=$gpu)
script_params+=(--num_gpus 1)
fi
mkdir -p $output_dir
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment