How can I make a custom metric that return more than 1 number?

I am able to run custom evaluations of my model in this notebook. Hope it helps you.

I will continue tomorrow, I just copy the whole lm_rnn to my notebook so it is easier to add pdb.set_trace() to debug. I run lr_find() and I have self.hidden and able to run m(*VV(xs)) when I was inside lr_find(), not sure why it is missing outside the function…

the self.hidden attribute was created by reset(), I notice that by printing a checkpoint inside the function, it prints when I do lr_find() but not when I am evaluating a batch.

Turns out Jeremy actually answer my question already… but I did not see it.

1 Like