type: MSELoss
args:
  reduction: mean