__init__(self, model, input_record, name="adaptive_weight", optimizer=None, weights=None, enable_diagnose=False, estimation_method="log_std", pos_optim_method="log_barrier", reg_lambda=0.1, kwargs) (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
caffe2::python::layers::layers::ModelLayer.__init__(self, model, prefix, input_record, predict_input_record_fields=None, tags=None, kwargs) | caffe2.python.layers.layers.ModelLayer | |
add_eval_ops(self, net) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
add_init_params(self, init_net) | caffe2.python.layers.layers.ModelLayer | |
add_operators(self, net, init_net=None, context=InstantiationContext.TRAINING) | caffe2.python.layers.layers.ModelLayer | |
add_ops(self, net) (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
add_ops_to_accumulate_pred(self, net) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
add_param_copy_operators(self, net) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
add_train_ops(self, net) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
concat_data(self, net) (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
create_param(self, param_name, shape, initializer, optimizer, ps_param=None, regularizer=None) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
data (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
enable_diagnose (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
estimation_method (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
eval_output_schema (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
export_output_for_metrics(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
export_params_for_metrics(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
get_fp16_compatible_parameters(self) | caffe2.python.layers.layers.ModelLayer | |
get_memory_usage(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
get_next_blob_reference(self, name) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
get_parameters(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
get_type(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
init_func (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
input_record(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
inv_var_init(self) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
inv_var_reg(self, net, reg) (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
inv_var_weight(self, x, net, weight) (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
k (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
kwargs (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
log_std_init(self) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
log_std_reg(self, net, reg) (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
log_std_weight(self, x, net, weight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
model (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
mu (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
name (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
num (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
optimizer (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
output_schema (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
output_schema(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
output_schema(self, output_schema) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
params (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
pos_optim_method (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
precomputation_object_only (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
precomputation_request_only (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
predict_input_record(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
predict_output_schema(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
predict_output_schema(self, output_schema) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
reg_func (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
reg_lambda (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
request_only (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
tags (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
weight_func (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
weight_i (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |
weights (defined in caffe2.python.layers.adaptive_weight.AdaptiveWeight) | caffe2.python.layers.adaptive_weight.AdaptiveWeight | |