1  import torch
2  import torch.nn.functional as F
3  
4  def box_loss(output_box,target_box):
5      return F.smooth_l1_loss(output_box,target_box)