generate_train_pickle.py 1.5 KB