Hello everyone, Using tensorrt for custom yolo training, with Quadro RTX GPU, 1 class dataset and ~40k images, I run into a lot of memory trouble. First it was cpu memory fail, then cuda host allocation fails, depending on my batch_size. Here is my config : random_seed: 42 yolo_config { big_an…

Please check if below link can help. [image] OOM Error after 10 epochs in SSD Resnet-18 TAO Toolkit Hi psgr, Your original mentioned issue is from evaluation instead of training, right? If for evaluation, please set below $ export TF_FORCE_GPU_ALLOW_GROWTH=true From yo…

Custom training yolov3 out of mermory tlt

Accelerated Computing Intelligent Video Analytics TAO Toolkit

Morganh October 2, 2020, 5:27pm 7

Please check if below link can help.