Try reducing batch_size attribute to a small number(like 1,2 or 3).
Example:
train_generator = data_generator.flow_from_directory(
'path_to_the_training_set',
target_size = (IMG_SIZE,IMG_SIZE),
batch_size = 2,
class_mode="categorical"
)
More Related Contents:
- Keras Dense layer’s input is not flattened
- Negative dimension size caused by subtracting 3 from 1 for ‘conv2d_2/convolution’
- How do I get the weights of a layer in Keras?
- How to fix “AttributeError: module ‘tensorflow’ has no attribute ‘get_default_graph'”?
- Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
- How to tell if tensorflow is using gpu acceleration from inside python shell?
- Using a pre-trained word embedding (word2vec or Glove) in TensorFlow
- What is the difference between ‘SAME’ and ‘VALID’ padding in tf.nn.max_pool of tensorflow?
- Memory errors and list limits?
- Get total physical memory in Python
- Tensorflow: How to get a tensor by name?
- TypeError: __array__() takes 1 positional argument but 2 were given (Image classification Keras)
- “zsh: illegal hardware instruction python” when installing Tensorflow on macbook pro M1 [duplicate]
- Tracking *maximum* memory usage by a Python function
- Measure Object Size Accurately in Python – Sys.GetSizeOf not functioning
- Deep version of sys.getsizeof [duplicate]
- Python in-memory zip library
- Keras Custom loss function to pass arguments other than y_true and y_pred
- TensorFlow: training on my own image
- Unknown initializer: GlorotUniform when loading Keras model
- How to find which version of TensorFlow is installed in my system?
- No Module Named ‘_pywrap_tensorflow_internal’
- Python deep getsizeof list with contents?
- Error while import tensorflow module
- Unexpected keyword argument ‘ragged’ in Keras
- Tensorflow and Multiprocessing: Passing Sessions
- Sparse Tensor (matrix) from a dense Tensor Tensorflow
- how to convert numpy to tfrecords and then generate batches?
- How to fix ipykernel_launcher.py: error: unrecognized arguments in jupyter?
- How to prefetch data using a custom python function in tensorflow