search

chenhsuanlin / inverse-compositional-STN

Inverse Compositional Spatial Transformer Networks :performing_arts: (CVPR 2017 oral)
MIT License
318 stars 63 forks source link

Stuck at loading MNIST dataset #5

Closed phongnhhn92 closed 7 years ago

phongnhhn92 commented 7 years ago

Hello guys, I tried to run train.py using this command python train.py CNN --group groupCNN --model modelCNN --lr 0.001 --lr 0.001 --batchSize 32 --maxIter 2000 --warpType translation But somehow I stuck at this loading MNIST dataset. My Internet works normally so I dont know what is going on. I also dont know if the parameter I used on the command above is correct or not since the readme doesnt provide any examples to compare. Thanks a

chenhsuanlin commented 7 years ago

The command line works fine on my side. Could you double-check whether there are files downloaded to data and MNIST_data? The code should automatically download MNIST if doesn't exist in your directory.