I am trying to fine tune MaskRCNN instance segmentation following this tutorial. My custom dataset and model build seem to pass, but I run into this ZeroDivisionError upon training. Any hints on how to debug are appreciated. Happy to include more code, did not want to overwhelm with irrelevant …


ZeroDivisionError                         Traceback (most recent call last)
 in ()
4 for epoch in range(num_epochs):
5     # train for one epoch, printing every 10 iterations
----> 6     train_one_epoch(model, optimizer, data_loader, device, epoch,print_freq=100)
7     # update the learning rate
8     lr_scheduler.step()
1 frames
/content/utils.py in log_every(self, iterable, print_freq, header)
216         total_time_str = str(datetime.timedelta(seconds=int(total_time)))
217         print(’{} Total time: {} ({:.4f} s / it)’.format(
 → 218             header, total_time_str, total_time / len(iterable)))
219
220
ZeroDivisionError: float division by zero