There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks
Above notebook is for google Colab. Instead, for running locally, please download notebook from TAO Toolkit Quick Start Guide - NVIDIA Docs → TAO Toolkit Getting Started | NVIDIA NGC