Use negative loss value as the key model validation metric

A new AuxiliaryOperation called NegLoss has been added to help facilitate using the minimal loss as the key metric.

In your config_train.json, add the following into the “train” section:

Copy
Copied!
            

"aux_ops": [ { "path": "NegLoss", "args": { "tag”: "negloss" } } ],

Then in the metric section of “validate”:

Copy
Copied!
            

"metrics": [ { "name": "ComputeAverage", "args": { "name": "negloss", "is_key_metric": true, "field": "negloss" } } ],

© Copyright 2020, NVIDIA. Last updated on Feb 2, 2023.