diff --git a/bin/cml-runner.js b/bin/cml-runner.js index 8e4cd2b17..6886709bf 100755 --- a/bin/cml-runner.js +++ b/bin/cml-runner.js @@ -161,7 +161,7 @@ const run_cloud = async (opts) => { cloud, region, type, - gpu: gpu.replace(/^tesla$/, 'v100'), + gpu: gpu === 'tesla' ? 'v100' : gpu, hdd_size, ssh_private, spot,