Python batch_atomic_generateの例

コード例 #1

ファイル: atomic_train.py プロジェクト: PlusLabNLP/R-3

    def batch(self, opt, *args):
        outputs = batch.batch_atomic_generate(opt, *args)

        token_loss = outputs["loss"]
        nums = outputs["nums"]
        reset = outputs["reset"]

        return token_loss, nums, reset

コード例 #2

ファイル: atomic_train.py プロジェクト: Heidelberg-NLP/COINS

    def batch(self, opt, *args):
        outputs = batch.batch_atomic_generate(opt, *args)

        token_loss_knowledge = outputs["loss_knowledge"]
        token_loss_sentence = outputs["loss_sentence"]
        nums = outputs["nums"]
        reset = outputs["reset"]

        return token_loss, nums, reset

コード例 #3

ファイル: conceptnet_train.py プロジェクト: Heidelberg-NLP/COINS

    def batch(self, opt, *args):
        outputs = batch_utils.batch_atomic_generate(opt, *args)

        token_loss_knowledge = outputs[
            "loss_knowledge"]  #+ outputs["loss_sentence"]
        token_loss_sentence = outputs[
            "loss_sentence"]  #+ outputs["loss_knowledge"]
        nums_s = outputs["nums_s"]
        nums_k = outputs["nums_k"]
        reset = outputs["reset"]

        return token_loss_knowledge, token_loss_sentence, nums_k, nums_s, reset

コード例 #4

ファイル: atomic_train.py プロジェクト: mnskim/teamproject

    def batch(self, opt, *args):
        #ipdb.set_trace()
        if opt['train']['static'].train_comet_loss:
            _func = functools.partial(batch.batch_atomic_generate,
                                      comet_loss=True)
            outputs = _func(opt, *args)
        else:
            outputs = batch.batch_atomic_generate(opt, *args)

        token_loss = outputs["loss"]
        nums = outputs["nums"]
        reset = outputs["reset"]

        return token_loss, nums, reset