basecls.layers.losses

basecls.layers.losses.build_loss(cfg)

The factory function to build loss.

参数

cfg (ConfigDict) – config for building loss function.

返回类型

Module

返回

A loss function.

class basecls.layers.losses.BinaryCrossEntropy(**kwargs)

基类：Module

The module for binary cross entropy.

See binary_cross_entropy() for more details.

forward(x, y)

返回类型

Tensor

class basecls.layers.losses.CrossEntropy(axis=1, label_smooth=0.0)

基类：Module

The module for cross entropy.

It supports both categorical labels and one-hot labels. See cross_entropy() for more details.

参数

• axis (int) – reduced axis. Default: 1

• label_smooth (float) – label smooth factor. Default: 0.0

forward(x, y)

返回类型

Tensor