Thanks for your reply. This is the original symbol.json file that I have (which should have the weights in it too)
{
“op”: “Custom”,
“name”: “softmax26”,
“attrs”: {
“class_weights”: “0.1,10”,
“op_type”: “weighted_softmax_ce_loss”
},
“inputs”: [[637, 0, 0], [638, 0, 0]]
},
{
“op”: “MakeLoss”,
“name”: “makeloss0”,
“inputs”: [[639, 0, 0]]
},
{
“op”: “_mul_scalar”,
“name”: “_mulscalar16”,
“attrs”: {“scalar”: “8”},
“inputs”: [[640, 0, 0]]
},
{
“op”: “null”,
“name”: “fc_t26_weight”,
“attrs”: {“num_hidden”: “2”},
“inputs”:
},
{
“op”: “null”,
“name”: “fc_t26_bias”,
“attrs”: {“num_hidden”: “2”},
“inputs”:
},
I have modified it to be like:
{
“op”: “SoftmaxOutput”,
“name”: “softmax26”,
“attrs”: {
},
“inputs”: [[637, 0, 0], [638, 0, 0]]
},
{
“op”: “MakeLoss”,
“name”: “makeloss0”,
“inputs”: [[639, 0, 0]]
},
{
“op”: “_mul_scalar”,
“name”: “_mulscalar16”,
“attrs”: {“scalar”: “8”},
“inputs”: [[640, 0, 0]]
},
{
“op”: “null”,
“name”: “fc_t26_weight”,
“attrs”: {“num_hidden”: “2”},
“inputs”:
},
{
“op”: “null”,
“name”: “fc_t26_bias”,
“attrs”: {“num_hidden”: “2”},
“inputs”:
},
is that what you meants?