autokeras icon indicating copy to clipboard operation
autokeras copied to clipboard

`resnet_v2`-models can cause out of memory error

Open Anselmoo opened this issue 2 years ago • 1 comments

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

Anselmoo avatar May 13 '22 20:05 Anselmoo

See also:

https://github.com/keras-team/autokeras/blob/master/autokeras/utils/utils.py#L87

Anselmoo avatar May 13 '22 20:05 Anselmoo