rm -r models
rm -r pretrain