frame_decoration

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