Я использую 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
?
Примечание. Я использую две параллельные сети, и эти параметры являются суммой параметров обеих сетей.