autokeras
autokeras copied to clipboard
`resnet_v2`-models can cause out of memory error
Bug Description
In the case of using resnet_v2
-models, an out-of-memory error can be generated.
Expected Behavior
Instead of crashing and generating an out-of-memory error, the currently used model should be skipped.
Setup Details
Include the details about the versions of:
- OS type and version: Ubuntu 20
- Python: 3.8
- autokeras: latest version
Additional context
See also discussion: #1721
See also:
https://github.com/keras-team/autokeras/blob/master/autokeras/utils/utils.py#L87