Tag: PyTorch

  • PyTorch Lightning: How To Keep Your PyTorch Project Clean

    As an engineer in perception and computer vision, I tend to work with PyTorch a lot.  It’s a very flexible framework and there are already many useful components built into it that prevent you from having to build it yourself.  However, if you have already built a few projects with it, you might have noticed…

  • Why You Should Always Visualize A Batch From Your Data Loader

    Here is one thing that every engineer working with AI hears at some point: Garbage In, Garbage Out However, despite that, something that I still see often enough in the Computer Vision industry and no doubt affects other machine learning disciplines as well is a complete lack of checking what is coming out of the…

  • Einops: Making Life Bit Easier (Mostly)

    When working in PyTorch, you are often faced with the need to manipulate tensors of multiple dimensions into various shapes or maybe combine dimensions together.  There’s a wealth of functions specifically for that too.  To name a few, view, permutate, stack, tile, and concat are some of the most common ones, but the list goes…

en_USEN