Skip to content

where does the mmi loss function locate? #8

@lsy641

Description

@lsy641

I just see two losscompute in train.py
`def make_loss_compute(model, tgt_vocab, opt):

if opt.copy_attn:
    compute = onmt.modules.CopyGeneratorLossCompute(
        model.generator, tgt_vocab, opt.copy_attn_force)
else:
    compute = onmt.Loss.NMTLossCompute(
        model.generator, tgt_vocab,
        label_smoothing=opt.label_smoothing)

if use_gpu(opt):
    compute.cuda()

return compute`

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions