- 000
Some popular image models like Mask-RCNN for object detection calculate the loss in a slightly different way:
loss_dict = model(images, targets)
losses = sum(loss for loss in loss_dict.values())
other than
out = model(images)
loss = criteria(out, targets)
Orca estimator wants to support this model....