search

itayhubara / BinaryNet.tf

BNN implementation in tensorflow
166 stars 55 forks source link

Resource exhausted: OOM when allocating tensor with shape[4608,1024] #6

Open jeremy-myers opened 6 years ago

jeremy-myers commented 6 years ago

When I evaluate the sample, I attempt to pass the argument --batch_size = N for a range of N, but I receive this error 2018-05-01 22:46:06.579010: W tensorflow/core/framework/op_kernel.cc:1158] Resource exhausted: OOM when allocating tensor with shape[3,3,128,256]

In my opinion, it doesn't seem like the code is responsive to my argument. What is the correct way to change the batch size for BNN_cifar10?

cw-plus commented 6 years ago

In the code:def train(model, data, batch_size=128, learning_rate=FLAGS.learning_rate, log_dir='./log', checkpoint_dir='./checkpoint', num_epochs=-1): seems batchsize not get from

--batch_size = N

So, I recommend you change it in train function.