1

tiny-imagenet私の Google colab は、10 万枚の画像とサイズ 64*64 の 200 クラスでトレーニングを開始するとすぐにクラッシュします。

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まで試行)で使用していますが、何も機能しません。

(1.8*10^9 パラメータ) のアーキテクチャと、80 万パラメータのカスタム モデルを試してみresnet-18ましたが、無駄でした。

他の情報が必要な場合に備えて、colab へのリンクを貼り付け ます

4

0 に答える 0