Optimizing PyTorch Performance: Batch Size with PyTorch Profiler
Briefly