mmagic.models.editors.liif.mlp_refiner

Module Contents

Classes

MLPRefiner

Multilayer perceptrons (MLPs), refiner used in LIIF.

class mmagic.models.editors.liif.mlp_refiner.MLPRefiner(in_dim, out_dim, hidden_list)

Bases: mmengine.model.BaseModule

Multilayer perceptrons (MLPs), refiner used in LIIF.

参数

• in_dim (int) – Input dimension.

• out_dim (int) – Output dimension.

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

forward(x)

Forward function.

参数

x (Tensor) – The input of MLP.

返回

The output of MLP.

返回类型

Tensor