Skip to content

GitLab

Unverified Commit dfe3c27b authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Fix aml v2 config (#3655)

parent 8f002c8b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
ts/nni_manager/training_service/reusable/environments/amlEnvironmentService.ts
View file @ dfe3c27b
...@@ -101,7 +101,9 @@ export class AMLEnvironmentService extends EnvironmentService { ...@@ -101,7 +101,9 @@ export class AMLEnvironmentService extends EnvironmentService {
amlEnvironment.command = `mv envs outputs/envs && cd outputs && ${amlEnvironment.command}`; amlEnvironment.command = `mv envs outputs/envs && cd outputs && ${amlEnvironment.command}`;
} }
amlEnvironment.command = `import os\nos.system('${amlEnvironment.command}')`; amlEnvironment.command = `import os\nos.system('${amlEnvironment.command}')`;
amlEnvironment.useActiveGpu = !!this.config.deprecated.useActiveGpu; if (this.config.deprecated && this.config.deprecated.useActiveGpu !== undefined) {
amlEnvironment.useActiveGpu = this.config.deprecated.useActiveGpu;
}
amlEnvironment.maxTrialNumberPerGpu = this.config.maxTrialNumberPerGpu; amlEnvironment.maxTrialNumberPerGpu = this.config.maxTrialNumberPerGpu;
await fs.promises.writeFile(path.join(environmentLocalTempFolder, 'nni_script.py'), amlEnvironment.command, { encoding: 'utf8' }); await fs.promises.writeFile(path.join(environmentLocalTempFolder, 'nni_script.py'), amlEnvironment.command, { encoding: 'utf8' });
......
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