Shortcuts

mmedit.models.editors.liif.mlp_refiner

Module Contents

Classes

MLPRefiner

Multilayer perceptrons (MLPs), refiner used in LIIF.

class mmedit.models.editors.liif.mlp_refiner.MLPRefiner(in_dim, out_dim, hidden_list)[source]

Bases: mmengine.model.BaseModule

Multilayer perceptrons (MLPs), refiner used in LIIF.

Parameters
  • in_dim (int) – Input dimension.

  • out_dim (int) – Output dimension.

  • hidden_list (list[int]) – List of hidden dimensions.

forward(x)[source]

Forward function.

Parameters

x (Tensor) – The input of MLP.

Returns

The output of MLP.

Return type

Tensor

Read the Docs v: latest
Versions
master
latest
stable
zyh-re-docs
zyh-doc-notfound-extend
zyh-api-rendering
Downloads
pdf
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.