зеркало из https://github.com/microsoft/nni.git
Fix pipeline (#3994)
This commit is contained in:
Родитель
b42f85f5e1
Коммит
703f6de58c
|
@ -5,9 +5,6 @@ schedules:
|
|||
branches:
|
||||
include: [ master ]
|
||||
|
||||
variables:
|
||||
worker: remote_nni-ci-gpu-03
|
||||
|
||||
# variables set on VSO: (for security concern)
|
||||
# manager_ip
|
||||
# worker_ip
|
||||
|
|
|
@ -24,6 +24,7 @@ jobs:
|
|||
|
||||
- script: |
|
||||
set -e
|
||||
export PATH="$PATH:/home/nni-test/ENTER/bin"
|
||||
python3 setup.py build_ts
|
||||
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]
|
||||
|
|
|
@ -33,6 +33,6 @@ python3 search.py --search-for micro --epochs 1
|
|||
#cd $EXAMPLE_DIR/naive
|
||||
#python3 train.py
|
||||
|
||||
echo "testing 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
|
||||
#echo "testing 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
|
||||
|
|
|
@ -19,7 +19,7 @@ import { ExperimentConfig, toSeconds, toCudaVisibleDevices } from '../common/exp
|
|||
import { ExperimentManager } from '../common/experimentManager';
|
||||
import { TensorboardManager } from '../common/tensorboardManager';
|
||||
import {
|
||||
TrainingService, TrialJobApplicationForm, TrialJobDetail, TrialJobMetric, TrialJobStatus, TrialCommandContent
|
||||
TrainingService, TrialJobApplicationForm, TrialJobDetail, TrialJobMetric, TrialJobStatus, TrialCommandContent, PlacementConstraint
|
||||
} from '../common/trainingService';
|
||||
import { delay, getCheckpointDir, getExperimentRootDir, getLogDir, getMsgDispatcherCommand, mkDirP, getTunerProc, getLogLevel, isAlive, killPid } from '../common/utils';
|
||||
import {
|
||||
|
@ -804,13 +804,14 @@ class NNIManager implements Manager {
|
|||
this.setStatus('RUNNING');
|
||||
}
|
||||
const trialRequestContent: TrialCommandContent = JSON.parse(content);
|
||||
const noneConstraint: PlacementConstraint = {type: 'None', gpus: []};
|
||||
const form: TrialJobApplicationForm = {
|
||||
sequenceId: this.experimentProfile.nextSequenceId++,
|
||||
hyperParameters: {
|
||||
value: content,
|
||||
index: 0
|
||||
},
|
||||
placementConstraint: trialRequestContent.placement_constraint
|
||||
placementConstraint: trialRequestContent.placement_constraint? trialRequestContent.placement_constraint : noneConstraint
|
||||
};
|
||||
this.waitingTrials.push(form);
|
||||
break;
|
||||
|
|
Загрузка…
Ссылка в новой задаче