This commit is contained in:
QuanluZhang 2021-07-29 23:13:23 +08:00 коммит произвёл GitHub
Родитель b42f85f5e1
Коммит 703f6de58c
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
4 изменённых файлов: 7 добавлений и 8 удалений

Просмотреть файл

@ -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;