Skip to content

resnet_v2-models can cause out of memory error  #1722

Open
@Anselmoo

Description

@Anselmoo

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions