train the project

[Jacobyan1]

您好，我最近在尝试运行您的代码，出现了 File "G:\YKZ\ISSAFE-master\train.py", line 105, in training
self.writer.add_scalar('train/total_loss_epoch', train_loss/num_img_tr, epoch)
~~~~~~~~~~^~~~~~~~~~~
ZeroDivisionError: float division by zero，我是一名新生，请问这个问题是出现在了数据集部分吗，已经困扰了我两三个月了