You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi,
I got an error from changing the backend to theano.
my command is :
python train.py --path ../Image\ Generation/ --TYPE "train" --batch_size 10 --epochs 10
Traceback (most recent call last):
File "train.py", line 196, in <module>
train(path = args.path, batch_size = args.batch_size, EPOCHS = args.epochs)
File "train.py", line 100, in train
generated_images = generator.predict(Noise_batch)
File "/home/akash/anaconda2/lib/python2.7/site-packages/keras/models.py", line 899, in predict
return self.model.predict(x, batch_size=batch_size, verbose=verbose)
File "/home/akash/anaconda2/lib/python2.7/site-packages/keras/engine/training.py", line 1573, in predict
batch_size=batch_size, verbose=verbose)
File "/home/akash/anaconda2/lib/python2.7/site-packages/keras/engine/training.py", line 1203, in _predict_loop
batch_outs = f(ins_batch)
File "/home/akash/anaconda2/lib/python2.7/site-packages/keras/backend/theano_backend.py", line 1122, in __call__
return self.function(*inputs)
File "/home/akash/anaconda2/lib/python2.7/site-packages/theano/compile/function_module.py", line 871, in __call__
storage_map=getattr(self.fn, 'storage_map', None))
File "/home/akash/anaconda2/lib/python2.7/site-packages/theano/gof/link.py", line 314, in raise_with_op
reraise(exc_type, exc_value, exc_trace)
File "/home/akash/anaconda2/lib/python2.7/site-packages/theano/compile/function_module.py", line 859, in __call__
outputs = self.fn()
RuntimeError: CorrMM failed to allocate working memory of 12800 x 65536
Apply node that caused the error: CorrMM{half, (1, 1)}(InplaceDimShuffle{0,3,1,2}.0, Subtensor{::, ::, ::int64, ::int64}.0)
Toposort index: 144
Inputs types: [TensorType(float32, 4D), TensorType(float32, 4D)]
Inputs shapes: [(10, 512, 4096, 16), (256, 512, 5, 5)]
Inputs strides: [(134217728, 4, 32768, 2048), (4, 1024, -2621440, -524288)]
Inputs values: ['not shown', 'not shown']
Outputs clients: [[InplaceDimShuffle{0,2,3,1}(CorrMM{half, (1, 1)}.0)]]
Backtrace when the node is created(use Theano flag traceback.limit=N to make it longer):
File "train.py", line 196, in <module>
train(path = args.path, batch_size = args.batch_size, EPOCHS = args.epochs)
File "train.py", line 63, in train
generator = model.generator_model()
File "/home/akash/GAN-master/model.py", line 34, in generator_model
model.add(Convolution2D(256, 5, 5, border_mode='same'))
File "/home/akash/anaconda2/lib/python2.7/site-packages/keras/models.py", line 463, in add
output_tensor = layer(self.outputs[0])
File "/home/akash/anaconda2/lib/python2.7/site-packages/keras/engine/topology.py", line 578, in __call__
output = self.call(inputs, **kwargs)
File "/home/akash/anaconda2/lib/python2.7/site-packages/keras/layers/convolutional.py", line 164, in call
dilation_rate=self.dilation_rate)
File "/home/akash/anaconda2/lib/python2.7/site-packages/keras/backend/theano_backend.py", line 1769, in conv2d
filter_dilation=dilation_rate)
HINT: Use the Theano flag 'exception_verbosity=high' for a debugprint and storage map footprint of this apply node.
The text was updated successfully, but these errors were encountered:
Hi,
I got an error from changing the backend to theano.
my command is :
python train.py --path ../Image\ Generation/ --TYPE "train" --batch_size 10 --epochs 10
The text was updated successfully, but these errors were encountered: