lossfx.h

File Path: modules/lossfx/include/templates/lossfx.h

File Type: H (Header)

Lines: 227

Dependencies

Includes:

• map

• notification/notification.h

• string

• structs/enums.h

• structs/optimizer.h

• tools/tools.h

• torch/torch.h

• vector

Classes

lossfx

Inherits from: ``tools,

public notification``

Methods:

• loss_enum loss_string(std::string name)

• opt_enum optim_string(std::string name)

• scheduler_enum scheduler_string(std::string name)

• void loss_opt_string(std::string name)

• Tensor loss(torch::Tensor* pred, torch::Tensor* truth)

• Tensor loss(torch::Tensor* pred, torch::Tensor* truth, loss_en...)

• void weight_init(torch::nn::Sequential* data, mlp_init method)

• void build_scheduler(optimizer_params_t* op, torch::optim::Optimizer* o...)

• bool build_loss_function(loss_enum lss)

• bool build_loss_function()

Functions

void _dress_reduction(g* imx, loss_opt* params)

void _dress_batch(g* imx, loss_opt* params)

void _dress_ignore(g* imx, loss_opt* params)

void _dress_smoothing(g* imx, loss_opt* params)

void _dress_margin(g* imx, loss_opt* params)

void _dress_blank(g* imx, loss_opt* params)

void _dress_zero(g* imx, loss_opt* params)

void _dress_swap(g* imx, loss_opt* params)

void _dress_eps(g* imx, loss_opt* params)

void _dress_beta(g* imx, loss_opt* params)

void _dress_full(g* imx, loss_opt* params)

void _dress_target(g* imx, loss_opt* params)

void _dress_delta(g* imx, loss_opt* params)