I couldn't find the outputs variable in this file in my docker image nvcr.io/nvidia/tao/tao-toolkit:5.0.0-tf1.15.5

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Continuing the discussion from Change prediction layer of pretrained model TAO:

Please provide the following information when requesting support.

• Hardware (T4/V100/Xavier/Nano/etc)
• Network Type (Detectnet_v2/Faster_rcnn/Yolo_v4/LPRnet/Mask_rcnn/Classification/etc)
• TLT Version (Please run “tlt info --verbose” and share “docker_tag” here)
• Training spec file(If have, please share here)
• How to reproduce the issue ? (This is for errors. Please share the command line and the detailed log here.)

2

Do you mean outputs = Dense(42, activation="softmax", name="predictions")(x) ?
That is the workaround code mentioned in that forum topic. It is not available in 5.0.0-tf1.15.5 docker.