Can't reproduce the example in the docs

[nidhaloff]

Bug Description

I'm just trying to reproduce the ImageClassification example. This is the error I receive. It just prints killed at the end. Do you have any idea? Is this problem specific to my machine?

2021-10-07 23:20:05.919840: W tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of 134217728 exceeds 10% of free system memory.
2021-10-07 23:20:06.146251: W tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of 134217728 exceeds 10% of free system memory.
2/2 [==============================] - 51s 26s/step - loss: 3.2675 - accuracy: 0.0781 - val_loss: 2.7888 - val_accuracy: 0.1667
Trial 1 Complete [00h 02m 03s]
val_loss: 2.7887697219848633

Best val_loss So Far: 2.7887697219848633
Total elapsed time: 00h 02m 03s

Search: Running Trial #2

Hyperparameter    |Value             |Best Value So Far 
image_block_1/b...|efficient         |resnet            
image_block_1/n...|True              |True              
image_block_1/a...|True              |True              
image_block_1/i...|True              |True              
image_block_1/i...|False             |True              
image_block_1/i...|0                 |0                 
image_block_1/i...|0                 |0                 
image_block_1/i...|0.1               |0.1               
image_block_1/i...|0                 |0                 
image_block_1/e...|True              |None              
image_block_1/e...|b7                |None              
image_block_1/e...|True              |None              
image_block_1/e...|True              |None              
classification_...|global_avg        |global_avg        
classification_...|0                 |0                 
optimizer         |adam              |adam              
learning_rate     |2e-05             |0.001             

Downloading data from https://storage.googleapis.com/keras-applications/efficientnetb7_notop.h5
258080768/258076736 [==============================] - 24s 0us/step
Killed

Bug Reproduction

As I posted, just tried to reproduce the example from the docs.

Data used by the code: mnist data as stated in the example

Expected Behavior

I just wanted to train an autokeras model using the mnist dataset.

PS: I'm using ubuntu 20.04