Custom Loss in Fastai

[abhikjha (Abhik)]

I am following what’s written here by Jeremy:

Where is the loss function used for a learner defined?

cool thanks @bushaev!

and one more post here:

How to make a custom loss function (PyTorch)

I hadn’t seen that yet actually so thank you. It is helpful but doesn’t looks like it solves my issue. I can define a working function, the problem is when i try to assign it to my model.crit= MY_NEW_FUNCTION is through an error looking for inputs into the function that aren’t calculated yet. def weighted_mse_loss(input, target): weight = torch.ones(16) weight[0] = 3 return torch.mean(weight * (input - target) ** 2) m.crit=weighted_mse_loss() gives me this error: ---------------…