Tag: machine learning
-
Should Machine Learning Engineers Write Unit Tests?
If you are looking for a tl;dr, the answer is yes. I somehow seem to run into engineers fairly frequently that think there is nothing to test in machine learning code or at least can’t imagine where one might need a test. Generally speaking, the argument is something like “It’s just a model. You just…
-
Training on GPU with Libtorch and How to Use MPS
Arguably the most useful feature of deep learning libraries is their ability to move workloads to the GPU and PyTorch makes it very easy for users by providing a to() function that you can pass a device object to. The documentation doesn’t make it very clear about it, but actually Libtorch makes it just as…
-
Metric Tracking Options for Libtorch
For anyone reading articles on this website, I don’t think I need to stress the importance of logging metrics of ML models, but there aren’t very many options when it comes to Libtorch metric tracking. Things just aren’t quite as easy as they are with PyTorch. You could, of course, save everything to a csv…
-
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: Make Your Life 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…
-
Setting up Libtorch in a Simple Docker Environment
In this article, let’s take a look on how to setup docker for Libtorch projects. Environment management is an important topic even for machine learning projects, so don’t get caught without a Dockerfile in your project! A couple of prerequisites: First, Let’s Setup Our Virtual Environment Whether you are programming in Python, C++, or really…
-
Getting Started with Libtorch
In the interest of getting as many readers as fast as possible, for the first post on this blog, I’m going to jump straight into the most exciting topic I can think of: C++. Sorry, not sorry. In the future, I’ll be writing about everything from computer perception and machine learning to more general software…
EN 
ES