pytorch
ae285d83 - [1/n][caffe2] Add session based margin loss function in caffe2 operator

Commit
2 years ago
[1/n][caffe2] Add session based margin loss function in caffe2 operator Summary: Add session based margin loss into caffe2 operator. This is the first diff make these 2 loss available to dper3 Test Plan: unit test succeeds with gradient check for both new loss function buck test //caffe2/caffe2/python/operator_test:softmax_l2r_operator_test buck test //caffe2/caffe2/python/operator_test:margin_loss_l2r_operator_test E2E test in bento notebook with model training in N1488923 margin loss model: f318207967 f318207399 Notice that the E2E test is run with dper change in D33532976 to change a full model Reviewed By: devashisht Differential Revision: D32902460 fbshipit-source-id: 8f21b9109f500583431156908b632e503ed90dbd (cherry picked from commit 1592111aa4ed6cfdd7ca5f54de70396e9610757c)
Author
Ziheng Huang
Committer
Parents
Loading