Shortcuts

Conv2dRFSearchOp

class mmcv.cnn.Conv2dRFSearchOp(op_layer: torch.nn.modules.module.Module, global_config: dict, verbose: bool = True)[源代码]

Enable Conv2d with receptive field searching ability.

参数
  • op_layer (nn.Module) – pytorch module, e,g, Conv2d

  • global_config (dict) –

    config dict. Defaults to None. By default this must include:

    • ”init_alphas”: The value for initializing weights of each branch.

    • ”num_branches”: The controller of the size of search space (the number of branches).

    • ”exp_rate”: The controller of the sparsity of search space.

    • ”mmin”: The minimum dilation rate.

    • ”mmax”: The maximum dilation rate.

    Extra keys may exist, but are used by RFSearchHook, e.g., “step”, “max_step”, “search_interval”, and “skip_layer”.

  • verbose (bool) – Determines whether to print rf-next related logging messages. Defaults to True.

estimate_rates()None[源代码]

Estimate new dilation rate based on trained branch_weights.

expand_rates()None[源代码]

Expand dilation rate.

forward(input: torch.Tensor)torch.Tensor[源代码]

Defines the computation performed at every call.

Should be overridden by all subclasses.

注解

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

Read the Docs v: stable
Versions
latest
stable
2.x
v2.0.1
v2.0.0
1.x
v1.7.1
v1.7.0
v1.6.2
v1.6.1
v1.6.0
v1.5.3
v1.5.2_a
v1.5.1
v1.5.0
v1.4.8
v1.4.7
v1.4.6
v1.4.5
v1.4.4
v1.4.3
v1.4.2
v1.4.1
v1.4.0
v1.3.18
v1.3.17
v1.3.16
v1.3.15
v1.3.14
v1.3.13
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.