进行代码,摘自
安装
export CUDA_ROOT=...
make
mkdir mnist
cd mnist
wget www.cs.toronto.edu/~nitish/data/mnist.h5
mkdir models
cd ..
跑步
训练
python lstm_composite.py lstm_combo_1layer_mnist.pbtxt bouncing_mnist.pbtxt
最终我得到了nan的分数运行时
Step 17000 Dec 213.17694 Fut 345.98576 Writing model to mnist/models/lstm_autoencoder_20150911163609.h5
Step 17100 Dec 213.55469 Fut 347.48344
Step 17200 Dec 213.18424 Fut 346.63
评论0
最新资源