886e263efd4c0ad85a6c462e668a26177d0a8716,onmt/modules/StructuredAttention.py,,,#,44

Before Change

if __name__ == "__main__":
    logging = onmt.io.IO.set_logger("StructuredAttention.py")
    dtree = MatrixTree()
    q = torch.rand(1, 5, 5).cuda()
    marg = dtree.forward(q)
    logging.info(marg.sum(1))

After Change

if __name__ == "__main__":
    logger = get_logger("StructuredAttention.log")
    dtree = MatrixTree()
    q = torch.rand(1, 5, 5).cuda()
    marg = dtree.forward(q)
    logger.info(marg.sum(1))

Instances