bcnn_DD_woft.py error - Githubissues

abhaydoke09 / Bilinear-CNN-TensorFlow

This is an implementation of Bilinear CNN for fine grained visual recognition using TensorFlow.

191 stars 72 forks source link

bcnn_DD_woft.py error #32

Closed MBB001 closed 5 years ago

MBB001 commented 6 years ago

i meet an error： Data shapes -- (train, val, test) (3300, 488, 488, 3) (330, 224, 224, 3) Traceback (most recent call last): File "bcnn_DD_woft.py", line 286, in X_train, Y_train = shuffle(X_train, Y_train) File "/home/mb/.local/lib/python2.7/site-packages/tflearn/data_utils.py", line 639, in shuffle arrs[i] = np.array(arr) MemoryError who can tell me what should i do. thank you!

abhaydoke09 commented 5 years ago

@MBB001

I think you might need to use CPU/GPU with more memory. I trained this model with 24GB GPU memory