PyTorch
PyTorch
PyTorch

PyTorch Model Conversion and Inference with ONNX

2023-03-05

PyTorch Model Conversion and Inference with ONNX

This article covers the process of converting a PyTorch model to ONNX format, verifying the converted model, and performing inference using the ONNX model. It also provides information on using pretrained ONNX models and inspecting the model layers.

Machine Learning
Machine Learning
ONNX
ONNX
PyTorch
PyTorch
torch.stack in PyTorch

2023-02-17

torch.stack in PyTorch

This article provides an overview of the torch.stack function in PyTorch and its syntax and parameters. It also covers the usage of the dim parameter in torch.stack with examples of how to change the parameter. Additionally, the article compares torch.stack with other PyTorch functions, such as torch.cat and torch.chunk, and discusses their differences with examples. Finally, the article addresses performance and memory considerations when using torch.stack in deep learning models, including the use of the in-place operation and the out parameter to reduce memory usage.

Python
Python
PyTorch
PyTorch
Convolutional Neural Network (CNN)

2022-10-27

Convolutional Neural Network (CNN)

This article explains Convolutional Neural Networks (CNNs), their architecture, and how to visualize their inner workings.

Machine Learning
Machine Learning
Deep Learning
Deep Learning
PyTorch
PyTorch