[Code] Updated train_classifier.py to remove the severe bug that was also present in the ondisk training implementation
[Code] Updated train_classifier.py to remove the severe bug that was also present in the ondisk training implementation

erfannoury/SuperEdge,erfannoury/SuperEdge,erfannoury/SuperEdge,erfannoury/SuperEdge,erfannoury/SuperEdge