Unet inference error

Hi,
Refer to Unet Inference Error
I realized that in jupyter notebook, when I go to test dataset directory, the .ipynb_checkpoints is generated, which leads to an error in unet inference.

I think it is better to manage this in TAO source code.

You can change the path where to trigger the notebook.