fixed match count in train.py