search

awjuliani / TF-Tutorials

A collection of deep learning tutorials using Tensorflow and Python
523 stars 189 forks source link

In deep network comparison #5

Open dotasheepliu opened 7 years ago

dotasheepliu commented 7 years ago

Hello! I followed the "Deep Network Comparison" tutorial: I put the DenseNet code block and all other code blocks (except for RegularNet, ResNet, and HighwayNet) into a python file, and tried to run it. However I see:

python densenet.py 
<IPython.core.display.HTML object>
Switched CIFAR set to 2
Step: 0 Loss: 2.64302 Accuracy: 0.109375
terminate called after throwing an instance of 'std::bad_alloc' 
  what():  std::bad_alloc

Did I run it in a wrong way? How should I solve this problem?

xuyifeng-nwpu commented 7 years ago

You should run this code in jupyter notebook ,but "python densenet.py"

Although I can run it correctly , the accuracy is still very low and the training time is over 10 hours.