xautodl/scripts/prepare.sh

14 lines
583 B
Bash
Raw Normal View History

2019-09-28 10:24:47 +02:00
#!/bin/bash
# bash ./scripts/prepare.sh
2020-01-09 12:26:23 +01:00
#datasets='cifar10 cifar100 imagenet-1k'
2019-09-28 10:24:47 +02:00
#ratios='0.5 0.8 0.9'
ratios='0.5'
save_dir=./.latent-data/splits
for ratio in ${ratios}
do
python ./exps/prepare.py --name cifar10 --root $TORCH_HOME/cifar.python --save ${save_dir}/cifar10-${ratio}.pth --ratio ${ratio}
python ./exps/prepare.py --name cifar100 --root $TORCH_HOME/cifar.python --save ${save_dir}/cifar100-${ratio}.pth --ratio ${ratio}
python ./exps/prepare.py --name imagenet-1k --root $TORCH_HOME/ILSVRC2012 --save ${save_dir}/imagenet-1k-${ratio}.pth --ratio ${ratio}
done