Pytorch Detectron Custom Dataset 1st problem

In this video I show that the Pytorch implementation of Detectron, found at https://github.com/roytseng-tw/Detectron.pytorch, will throw an error if you are trying to pass a dataset other than the ones specified in the train_net_step.py file. Commenting out the lines that throw this error is the solution.