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
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
Hide 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
.graph
import
*
from
.execution
import
*
...
...
nni/retiarii/codegen/pytorch.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
logging
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
"""
...
...
nni/retiarii/converter/graph_gen.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
re
import
torch
...
...
nni/retiarii/converter/op_types.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from
enum
import
Enum
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
):
if
isinstance
(
name
,
list
):
name
=
'__'
.
join
(
name
)
...
...
nni/retiarii/converter/visualize.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
graphviz
...
...
nni/retiarii/evaluator/functional.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
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.
import
abc
...
...
nni/retiarii/evaluator/pytorch/lightning.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import
warnings
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
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
os
import
random
...
...
@@ -63,13 +66,14 @@ class BaseExecutionEngine(AbstractExecutionEngine):
def
_send_trial_callback
(
self
,
paramater
:
dict
)
->
None
:
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
_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
:
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
:
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
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
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
.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
.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
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
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
logging
import
time
...
...
nni/retiarii/graph.py
View file @
e457047c
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
"""
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