Pranav Durai

icon indicating a website link https://www.linkedin.com/in/pranav-durai/ icon indicating an email [email protected]

icon company OpenCV icon location Bangalore, India icon location Sr. Computer Vision Engineer | Quantum AI Researcher | Research Fellow at Stanford University - School of Medicine

Results 4 comments of Pranav Durai

ISSUE: IndexError: list index out of range - while training 'YOLOv9-e' on custom dataset.

@Kitagawayyds @parasmaisalunkhe - Please go ahead with Ultralytics implementation of YOLOv9. Link: https://docs.ultralytics.com/models/yolov9/

ISSUE: IndexError: list index out of range - while training 'YOLOv9-e' on custom dataset.

@Kitagawayyds - Yes, I'm able to train both YOLOv9-C and E models on a custom dataset.

ISSUE: IndexError: list index out of range - while training 'YOLOv9-e' on custom dataset.

@parasmaisalunkhe - in that way, my dataset already had 640x640 images. It still didn't work.

Resource exhausted: OOM when allocating tensor with shape[32,960,10,10]

I was trying to train a **Vision Transformer on CIFAR-100 dataset**. GPU: **GTX 1650 w/ 4GB vRAM** Initially, I had the batch_size set to 256, which was totally insane for...