sohaib023 / siamese-pytorch

A PyTorch implementation of siamese networks using backbone from torchvision.models, with support for TensorRT inference.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Divided by zero

toppyishappy opened this issue · comments

There is some error with divided by zero because of empty losses list and initial total=0.
image

Its trying to divide by the number of training samples which in this case seem to be zero. I would add a check there, to circumvent the division by zero error, but for your purposes, I would suggest to double check the path and see if it has the images in the specified format.

It is my fault because the images need to be png.