python scripts/pretrain_in100.py

# CUB
sbatch generate_npy_CUB.sh
sbatch pretrain_CUB.sh