Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
703f6de5
Unverified
Commit
703f6de5
authored
Jul 29, 2021
by
QuanluZhang
Committed by
GitHub
Jul 29, 2021
Browse files
Fix pipeline (#3994)
parent
b42f85f5
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
7 additions
and
8 deletions
+7
-8
pipelines/integration-test-remote-l2l.yml
pipelines/integration-test-remote-l2l.yml
+0
-3
pipelines/integration-test-trt.yml
pipelines/integration-test-trt.yml
+1
-0
test/scripts/nas.sh
test/scripts/nas.sh
+3
-3
ts/nni_manager/core/nnimanager.ts
ts/nni_manager/core/nnimanager.ts
+3
-2
No files found.
pipelines/integration-test-remote-l2l.yml
View file @
703f6de5
...
@@ -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
...
...
pipelines/integration-test-trt.yml
View file @
703f6de5
...
@@ -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]
...
...
test/scripts/nas.sh
View file @
703f6de5
...
@@ -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
ts/nni_manager/core/nnimanager.ts
View file @
703f6de5
...
@@ -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
;
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment