ResourceExhaustedError при использовании Efficien tNet в керасе - PullRequest
1 голос
/ 28 февраля 2020

Я использую Google Colab. При использовании EfficientNetB3 я получаю следующую ошибку
Resource exhausted: OOM when allocating tensor with shape[15,95,95,192] and type float

Я понимаю это, потому что мои данные не помещаются в GPU. Но когда я пытаюсь InceptionResNetV2, я не получаю никакой ошибки.

Количество обучаемых параметров в EfficientNetB3 равно 22,220,824
Количество обучаемых параметров в InceptionResNetV2 равно 109,380,744

Количество обучаемых параметров в InceptionResNetV2 в 5 раз больше, чем EfficientNetB3. Поэтому я ожидаю, что InceptionResNetV2 выдаст ошибку не EfficientNetB3.

Любая идея, почему я получаю ошибку ресурса в EfficientNetB3?

Примечание. Я использую две параллельные сети, и эти параметры являются суммой параметров обеих сетей.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...