Please provide the following information when requesting support.
• Hardware (T4/V100/Xavier/Nano/etc) : Google Vertex AI
• Network Type (Detectnet_v2/Faster_rcnn/Yolo_v4/LPRnet/Mask_rcnn/Classification/etc) : Classification
• TLT Version (Please run “tlt info --verbose” and share “docker_tag” here): TAO 3.0
When attempting to run a custom job using the TAO container on google vertex ai platform, I receive the following error:
ImportError: cannot import name 'ONNXEngineBuilder'
Note I have received the same error when running the container locally and forget to use
--gpus all .
I was previously able to run training jobs on vertex ai platform using the TLT 2.0 container, and am trying to upgrade to TAO 3.0 but ran into this issue.
Any guidance is much appreciated.