next method
athulvingt opened this issue · 0 comments
athulvingt commented
Epoch 1 : Loss:0.0: 0%| | 0/1977 [00:00<?, ?it/s]
Traceback (most recent call last):
File "main.py", line 287, in
train(args.image, args.box2d, args.label)
File "main.py", line 172, in train
train_img, train_label = train_gen.next()
AttributeError: 'generator' object has no attribute 'next'