286a8cb1abde6e030e935734509d7f6832ce0f7f,OpenNMT/train.py,,trainModel,#Any#Any#Any#Any#Any#,154

Before Change

def trainModel(model, trainData, validData, dataset, info):
    params, gradParams = {}, {}

    def initParams(idx, args, state):
        // Only logs information of the first thread.
        verbose = idx == 0 and not opt.json_log

After Change

    for mod in model.values():
        mod.training()
        for p in mod.parameters():
            p.uniform_(-opt.param_init, opt.param_init)

    // define criterion of each GPU
    criterion = onmt.utils.Cuda.convert(buildCriterion(dataset.dicts.tgt.words.size(),
                                                       dataset.dicts.tgt.features))

    optim = onmt.train.Optim(

Instances