Unverified Commit 703f6de5 authored by QuanluZhang's avatar QuanluZhang Committed by GitHub
Browse files

Fix pipeline (#3994)

parent b42f85f5
...@@ -5,9 +5,6 @@ schedules: ...@@ -5,9 +5,6 @@ schedules:
branches: branches:
include: [ master ] include: [ master ]
variables:
worker: remote_nni-ci-gpu-03
# variables set on VSO: (for security concern) # variables set on VSO: (for security concern)
# manager_ip # manager_ip
# worker_ip # worker_ip
......
...@@ -24,6 +24,7 @@ jobs: ...@@ -24,6 +24,7 @@ jobs:
- script: | - script: |
set -e set -e
export PATH="$PATH:/home/nni-test/ENTER/bin"
python3 setup.py build_ts python3 setup.py build_ts
python3 setup.py bdist_wheel -p manylinux1_x86_64 python3 setup.py bdist_wheel -p manylinux1_x86_64
python3 -m pip install dist/nni-${NNI_RELEASE}-py3-none-manylinux1_x86_64.whl[SMAC,BOHB] python3 -m pip install dist/nni-${NNI_RELEASE}-py3-none-manylinux1_x86_64.whl[SMAC,BOHB]
......
...@@ -33,6 +33,6 @@ python3 search.py --search-for micro --epochs 1 ...@@ -33,6 +33,6 @@ python3 search.py --search-for micro --epochs 1
#cd $EXAMPLE_DIR/naive #cd $EXAMPLE_DIR/naive
#python3 train.py #python3 train.py
echo "testing pdarts..." #echo "testing pdarts..."
cd $EXAMPLE_DIR/legacy/pdarts #cd $EXAMPLE_DIR/legacy/pdarts
python3 search.py --epochs 1 --channels 4 --nodes 2 --log-frequency 10 --add_layers 0 --add_layers 1 --dropped_ops 3 --dropped_ops 3 #python3 search.py --epochs 1 --channels 4 --nodes 2 --log-frequency 10 --add_layers 0 --add_layers 1 --dropped_ops 3 --dropped_ops 3
...@@ -19,7 +19,7 @@ import { ExperimentConfig, toSeconds, toCudaVisibleDevices } from '../common/exp ...@@ -19,7 +19,7 @@ import { ExperimentConfig, toSeconds, toCudaVisibleDevices } from '../common/exp
import { ExperimentManager } from '../common/experimentManager'; import { ExperimentManager } from '../common/experimentManager';
import { TensorboardManager } from '../common/tensorboardManager'; import { TensorboardManager } from '../common/tensorboardManager';
import { import {
TrainingService, TrialJobApplicationForm, TrialJobDetail, TrialJobMetric, TrialJobStatus, TrialCommandContent TrainingService, TrialJobApplicationForm, TrialJobDetail, TrialJobMetric, TrialJobStatus, TrialCommandContent, PlacementConstraint
} from '../common/trainingService'; } from '../common/trainingService';
import { delay, getCheckpointDir, getExperimentRootDir, getLogDir, getMsgDispatcherCommand, mkDirP, getTunerProc, getLogLevel, isAlive, killPid } from '../common/utils'; import { delay, getCheckpointDir, getExperimentRootDir, getLogDir, getMsgDispatcherCommand, mkDirP, getTunerProc, getLogLevel, isAlive, killPid } from '../common/utils';
import { import {
...@@ -804,13 +804,14 @@ class NNIManager implements Manager { ...@@ -804,13 +804,14 @@ class NNIManager implements Manager {
this.setStatus('RUNNING'); this.setStatus('RUNNING');
} }
const trialRequestContent: TrialCommandContent = JSON.parse(content); const trialRequestContent: TrialCommandContent = JSON.parse(content);
const noneConstraint: PlacementConstraint = {type: 'None', gpus: []};
const form: TrialJobApplicationForm = { const form: TrialJobApplicationForm = {
sequenceId: this.experimentProfile.nextSequenceId++, sequenceId: this.experimentProfile.nextSequenceId++,
hyperParameters: { hyperParameters: {
value: content, value: content,
index: 0 index: 0
}, },
placementConstraint: trialRequestContent.placement_constraint placementConstraint: trialRequestContent.placement_constraint? trialRequestContent.placement_constraint : noneConstraint
}; };
this.waitingTrials.push(form); this.waitingTrials.push(form);
break; break;
......
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