Fix resume bug (#3569)

e0ec38e7 · SparkSnail · GitHub · 441267d1 · e0ec38e7
Unverified Commit e0ec38e7 authored Apr 23, 2021 by SparkSnail Committed by GitHub Apr 23, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

ts/nni_manager/core/nnimanager.ts ts/nni_manager/core/nnimanager.ts +5 -2

No files found.
--- a/ts/nni_manager/core/nnimanager.ts
+++ b/ts/nni_manager/core/nnimanager.ts
@@ -210,9 +210,12 @@ class NNIManager implements Manager {
            return Promise.resolve();
        }

-        this.log.info('Setup training service...');
        const config: ExperimentConfig = this.experimentProfile.params;
-        this.trainingService = await this.initTrainingService(config);
+        this.config = config;
+        if (this.trainingService === undefined) {
+            this.log.info('Setup training service...');
+            this.trainingService = await this.initTrainingService(config);
+        }

        this.log.info('Setup tuner...');
        const dispatcherCommand: string = getMsgDispatcherCommand(config);