File size: 268 Bytes
904ef7d |
1 2 3 4 5 6 7 8 9 10 11 |
import torch
import torch.nn as nn
import torch.nn.functional as F
def mape_loss(pred, target):
# pred, target: [B, 1], torch tenspr
difference = (pred - target).abs()
scale = 1 / (target.abs() + 1e-2)
loss = difference * scale
return loss.mean() |