Implementing your own Loss in TF JS

TensorFlow Special Interest Groups TF.js
,
1

Hi,

I am thinking of implementing YOLO in Node JS for learning purposes. I can see how the tfjs.layers will help, but I cant quite see how would you create a custom loss, or if it is possible.

Would this be using an Optimizer where the loss is my fn ?

Any help here?