CUDAのOut of Memory対策
連続して推論しようとすると1回目は通るが2回目以降で Out of memory が出る場合がある。
下記のように出力後に処理がある場合、使い終わったoutputはすぐにdelすると2回目以降が通るようになる。
while input in data:
output = model(input)
detections = process(output)
del output
モデルファイルを読み込む際のエラー対策
最終更新:2020年09月02日 23:33