Implementing custom augmentations online in tao model train

I would like to implement custom augmentations from albumentations like RandomResizedCrop to my tao training pipeline.

1. Is this possible?
2. If it is, can you point to to the source code where I can implement, or can you point me towards a similar post?

Thanks!

3 posts - 2 participants

Read full topic