-
Notifications
You must be signed in to change notification settings - Fork 1.8k
Conversation
import torch | ||
import torch.nn.functional as F | ||
import tensorflow as tf | ||
import nni.compression.tensorflow as tf_compressor | ||
import nni.compression.torch as torch_compressor |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this import order cannot pass pylint
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It can.
@@ -32,8 +32,8 @@ class AGP_Pruner(Pruner): | |||
""" | |||
def __init__(self, config_list): | |||
""" | |||
Configure Args | |||
initial_sparsity: | |||
Configure Args: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
please follow the docstring format like other tuners?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Updated.
Followed format used in msg_dispatcher.handle_report_metric_data
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think this format has some differences with the format that we used(We used the format like xgboost). I not sure about that. @leckie-chn
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This does not document the parameter list, instead it documents the fields of one parameter.
This reverts commit 97a4ed9.
Would you add a link to compressor docs in table of contents? Currently, there's no way to see compressor other than searching for it. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's postpone the known doc problem to next release, please create an issue to track it.
No description provided.