Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
e457047c
Unverified
Commit
e457047c
authored
Mar 09, 2021
by
QuanluZhang
Committed by
GitHub
Mar 09, 2021
Browse files
[retiarii] update debug info, and add license (#3438)
parent
539a7cd7
Changes
42
Show whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
64 additions
and
3 deletions
+64
-3
nni/retiarii/__init__.py
nni/retiarii/__init__.py
+3
-0
nni/retiarii/codegen/pytorch.py
nni/retiarii/codegen/pytorch.py
+3
-0
nni/retiarii/codegen/tensorflow.py
nni/retiarii/codegen/tensorflow.py
+3
-0
nni/retiarii/converter/graph_gen.py
nni/retiarii/converter/graph_gen.py
+3
-0
nni/retiarii/converter/op_types.py
nni/retiarii/converter/op_types.py
+3
-0
nni/retiarii/converter/utils.py
nni/retiarii/converter/utils.py
+3
-0
nni/retiarii/converter/visualize.py
nni/retiarii/converter/visualize.py
+3
-0
nni/retiarii/evaluator/functional.py
nni/retiarii/evaluator/functional.py
+3
-0
nni/retiarii/evaluator/pytorch/base.py
nni/retiarii/evaluator/pytorch/base.py
+3
-0
nni/retiarii/evaluator/pytorch/lightning.py
nni/retiarii/evaluator/pytorch/lightning.py
+3
-0
nni/retiarii/execution/api.py
nni/retiarii/execution/api.py
+3
-0
nni/retiarii/execution/base.py
nni/retiarii/execution/base.py
+7
-3
nni/retiarii/execution/cgo_engine.py
nni/retiarii/execution/cgo_engine.py
+3
-0
nni/retiarii/execution/interface.py
nni/retiarii/execution/interface.py
+3
-0
nni/retiarii/execution/listener.py
nni/retiarii/execution/listener.py
+3
-0
nni/retiarii/execution/logical_optimizer/interface.py
nni/retiarii/execution/logical_optimizer/interface.py
+3
-0
nni/retiarii/execution/logical_optimizer/logical_plan.py
nni/retiarii/execution/logical_optimizer/logical_plan.py
+3
-0
nni/retiarii/execution/logical_optimizer/opt_dedup_input.py
nni/retiarii/execution/logical_optimizer/opt_dedup_input.py
+3
-0
nni/retiarii/experiment/pytorch.py
nni/retiarii/experiment/pytorch.py
+3
-0
nni/retiarii/graph.py
nni/retiarii/graph.py
+3
-0
No files found.
nni/retiarii/__init__.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from
.operation
import
Operation
from
.operation
import
Operation
from
.graph
import
*
from
.graph
import
*
from
.execution
import
*
from
.execution
import
*
...
...
nni/retiarii/codegen/pytorch.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
logging
import
logging
from
typing
import
List
,
Tuple
,
Any
from
typing
import
List
,
Tuple
,
Any
...
...
nni/retiarii/codegen/tensorflow.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
# pylint: skip-file
# pylint: skip-file
"""
"""
...
...
nni/retiarii/converter/graph_gen.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
re
import
re
import
torch
import
torch
...
...
nni/retiarii/converter/op_types.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from
enum
import
Enum
from
enum
import
Enum
MODULE_EXCEPT_LIST
=
[
'Sequential'
]
MODULE_EXCEPT_LIST
=
[
'Sequential'
]
...
...
nni/retiarii/converter/utils.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
def
build_full_name
(
prefix
,
name
,
seq
=
None
):
def
build_full_name
(
prefix
,
name
,
seq
=
None
):
if
isinstance
(
name
,
list
):
if
isinstance
(
name
,
list
):
name
=
'__'
.
join
(
name
)
name
=
'__'
.
join
(
name
)
...
...
nni/retiarii/converter/visualize.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
graphviz
import
graphviz
...
...
nni/retiarii/evaluator/functional.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from
..graph
import
Evaluator
from
..graph
import
Evaluator
...
...
nni/retiarii/evaluator/pytorch/base.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
# This file is deprecated.
# This file is deprecated.
import
abc
import
abc
...
...
nni/retiarii/evaluator/pytorch/lightning.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
warnings
import
warnings
from
typing
import
Dict
,
Union
,
Optional
,
List
from
typing
import
Dict
,
Union
,
Optional
,
List
...
...
nni/retiarii/execution/api.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
time
import
time
from
..graph
import
Model
,
ModelStatus
from
..graph
import
Model
,
ModelStatus
...
...
nni/retiarii/execution/base.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
logging
import
logging
import
os
import
os
import
random
import
random
...
@@ -63,13 +66,14 @@ class BaseExecutionEngine(AbstractExecutionEngine):
...
@@ -63,13 +66,14 @@ class BaseExecutionEngine(AbstractExecutionEngine):
def
_send_trial_callback
(
self
,
paramater
:
dict
)
->
None
:
def
_send_trial_callback
(
self
,
paramater
:
dict
)
->
None
:
if
self
.
resources
<=
0
:
if
self
.
resources
<=
0
:
_logger
.
warning
(
'There is no available resource, but trial is submitted.'
)
# FIXME: should be a warning message here
_logger
.
debug
(
'There is no available resource, but trial is submitted.'
)
self
.
resources
-=
1
self
.
resources
-=
1
_logger
.
info
(
'Resource used. Remaining: %d'
,
self
.
resources
)
_logger
.
debug
(
'Resource used. Remaining: %d'
,
self
.
resources
)
def
_request_trial_jobs_callback
(
self
,
num_trials
:
int
)
->
None
:
def
_request_trial_jobs_callback
(
self
,
num_trials
:
int
)
->
None
:
self
.
resources
+=
num_trials
self
.
resources
+=
num_trials
_logger
.
info
(
'New resource available. Remaining: %d'
,
self
.
resources
)
_logger
.
debug
(
'New resource available. Remaining: %d'
,
self
.
resources
)
def
_trial_end_callback
(
self
,
trial_id
:
int
,
success
:
bool
)
->
None
:
def
_trial_end_callback
(
self
,
trial_id
:
int
,
success
:
bool
)
->
None
:
model
=
self
.
_running_models
[
trial_id
]
model
=
self
.
_running_models
[
trial_id
]
...
...
nni/retiarii/execution/cgo_engine.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
logging
import
logging
from
typing
import
List
,
Dict
,
Tuple
from
typing
import
List
,
Dict
,
Tuple
...
...
nni/retiarii/execution/interface.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from
abc
import
ABC
,
abstractmethod
,
abstractclassmethod
from
abc
import
ABC
,
abstractmethod
,
abstractclassmethod
from
typing
import
Any
,
NewType
,
List
,
Union
from
typing
import
Any
,
NewType
,
List
,
Union
...
...
nni/retiarii/execution/listener.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from
..graph
import
Model
,
ModelStatus
from
..graph
import
Model
,
ModelStatus
from
.interface
import
MetricData
,
AbstractGraphListener
from
.interface
import
MetricData
,
AbstractGraphListener
...
...
nni/retiarii/execution/logical_optimizer/interface.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from
abc
import
ABC
from
abc
import
ABC
from
.logical_plan
import
LogicalPlan
from
.logical_plan
import
LogicalPlan
...
...
nni/retiarii/execution/logical_optimizer/logical_plan.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
copy
import
copy
from
typing
import
Dict
,
Tuple
,
List
,
Any
from
typing
import
Dict
,
Tuple
,
List
,
Any
...
...
nni/retiarii/execution/logical_optimizer/opt_dedup_input.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from
typing
import
List
,
Dict
,
Tuple
from
typing
import
List
,
Dict
,
Tuple
from
nni.retiarii.utils
import
uid
from
nni.retiarii.utils
import
uid
...
...
nni/retiarii/experiment/pytorch.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
atexit
import
atexit
import
logging
import
logging
import
time
import
time
...
...
nni/retiarii/graph.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
"""
"""
Model representation.
Model representation.
"""
"""
...
...
Prev
1
2
3
Next
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