Блог

Главная - Вопросы по программированию - Google colab продолжает сбой при обучении модели (resnet18 / densenet) с использованием данных tiny-imagenet-200

Google colab продолжает сбой при обучении модели (resnet18 / densenet) с использованием данных tiny-imagenet-200

#conv-neural-network #google-colaboratory #imagenet

#conv-нейронная сеть #google-colaboratory #imagenet

Вопрос:

Мой Google colab выходит из строя сразу же, как только начинает обучение на tiny-imagenet с 0,1 миллионами изображений и 200 классами размером 64 * 64

Журнал Colab показывает

 WARNING:root:kernel 1fe0be22-c98a-4519-a16a-69c9fb4be1da restarted
KernelRestarter: restarting kernel (1/5), keep random ports 
tensorflow/stream_executor/dso_loader.cc:152] successfully opened CUDA library libcublas.so.10.0 locally
tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 10754 MB memory) -> physical GPU (device: 0, name: Tesla K80, pci bus id: 0000:00:04.0, compute capability: 3.7)

Я использую model.fit_generator с размером пакета (пробовал от 32 до 1024) и размером изображения (пробовал от 16 до 64), но ничего не работает.

Я пробовал resnet-18 архитектуру с параметрами (1,8 * 10 ^ 9), а также пользовательскую модель с 0,8 миллионами параметров, но тщетно.

Я вставляю ссылку на мой colab на случай, если кому-нибудь понадобится другая информацияhttps://colab.research.google.com/drive/1QG1mg1zOn6gZaaSv4rrI4F6erdxsxQ8V#scrollTo=Uy0M-VDHivOX