In pytorch_run.py 259 line, Why are you multiplying by x.size(0)

bigmb / Unet-Segmentation-Pytorch-Nest-of-Unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet

MIT License

1.87k stars 345 forks source link

Closed Y-P-W closed 4 years ago

bigmb commented 4 years ago

We use input images as a batch. So to calculate the loss we need to multiply it will the size of the batch.