原地址:https://github.com/Sarasra/models/tree/master/research/capsules
运行必须要有GPU
下载两个数据包,第一个是 cifar10 tfrecords,第二个是已经训练好的模型文件。
- Download and extract cifar10 binary version to $DATA_DIR/ from https://www.cs.toronto.edu/~kriz/cifar.html
- Download and extract cifar10 model checkpoints to $CKPT_DIR from:https://storage.googleapis.com/capsule_toronto/cifar_checkpoints.tar.gz
- Pass the directory that the binaries are extracted to ($DATA_DIR) as data_di
边做边写