moudsarewju

Hello, same bug when I use my own dataset ``` generator = image_util.ImageDataProvider("train/*.png") net = unet.Unet(channels=generator.channels, n_class=generator.n_class, layers=1, features_root=16) trainer = unet.Trainer(net, optimizer="momentum", opt_kwargs=dict(momentum=0.2)) path = trainer.train(generator, "./unet_trained", training_iters=20, epochs=100)...