Code

Support MMoE #1811

Code

Open

bluaxe wants to merge 9 commits intoTencent:dev_trainTencent/TNN:dev_trainfrom xudoong:dev_trainxudoong/TNN:dev_trainCopy head branch name to clipboard

+288-174

Commits

Commits on Sep 27, 2022

• Add support for multiple losses.

  xudoong

  committed
  2eec075

  Copy full SHA for 2eec075

• Fix the bug for RawBuffer allocation in InnerProduct.

  xudoong

  committed
  faead54

  Copy full SHA for faead54

• Change the default options of CMakeLists.txt

  xudoong

  committed
  fc64143

  Copy full SHA for fc64143

Commits on Sep 28, 2022

• Fix the loss_name push_back bug

  xudoong

  committed
  2bb975e

  Copy full SHA for 2bb975e

Commits on Oct 10, 2022

• Add arm grad op: add sub mul div.

  xudoong

  committed
  aacf19a

  Copy full SHA for aacf19a

• Fix typo in comment.

  xudoong

  committed
  0a1a545

  Copy full SHA for 0a1a545

Commits on Oct 25, 2022

• Merge remote-tracking branch 'upstream/dev_train' into dev_train

  xudoong

  committed
  ca6e200

  Copy full SHA for ca6e200

• Fix conflicts with upstream

  xudoong

  committed
  24f88d5

  Copy full SHA for 24f88d5

• Remove cout message in comments.

  xudoong

  committed
  5b254ab

  Copy full SHA for 5b254ab