train_cls.py 5.21 KB