When I run the "python train.py --saveto commoncraw_pretrained --dataset commoncrawl --cutoff 15", the got the following error:
Traceback (most recent call last):
File "train.py", line 341, in
train(**common_kwargs)
File "train.py", line 215, in train
cost = f_grad_shared(x, mask, ctx, cnn_feats)
File "/usr/local/lib/python2.7/dist-packages/theano/compile/function_module.py", line 786, in call
allow_downcast=s.allow_downcast)
File "/usr/local/lib/python2.7/dist-packages/theano/tensor/type.py", line 177, in filter
data.shape))
TypeError: ('Bad input argument to theano function with name "attention_generator/optimizers.py:64" at index 2(0-based)', 'Wrong number of dimensions: expected 3, got 2 with shape (256, 4500).')
How I solve it? Thanks!
Semi-Supervised-Image-Captioning copied to clipboard
wenhuchen
