Training with OCRNet 2.0 - Looking at old 1.0 jupyter notebook example

Hi everyone

I have configured OCRNet with TAO from this Jupyter notebook

tao_tutorials/notebooks/tao_launcher_starter_kit/ocrnet/ocrnet.ipynb at main · NVIDIA/tao_tutorials · GitHub

The notebook is using the version 1 of OCRNet, where the newest you can download is 2.1

I am using OCRNet trainable 2.0 and download it with ngc.
I also had to change the feature_channel: 512 tofeature_channel: 192
Running in nvcr.io/nvidia/tao/tao-toolkit:5.5.0-pyt

The notebook has the default epoch to be 10 which is fine verifying.

I am now training on the ICDAR15 dataset and I am seeing low accuracy. I am now running an experiment on 2000 epochs.

What are you others experience? What are good defaults? I tried searching on the forum and it there was not a lot of people talking about OCRNet with TAO 😄

2 posts - 1 participant

Read full topic