pytorch
creating your own pytorch scheduler
here is an example of a scheduler that I subclassed in pytorch.
pytorch
error fix: onnxruntime “Type Error: Type ‘tensor(int64)’ of input parameter of operator(Min) in node is invalid’
Background I was trying to convert pytorch bert-like model from torch to onnx and see if I can run it in onnxruntime. Here’s the environment I used python: 3.9 torch: 1.9.0 onnx: 1.10.2 onnxruntime-gpu: 1.9.0 I converted by pytorch model to onnx with the following line then I tried to Read more…
pytorch
is it okay to calculate loss in cpu and backpropagate in torch?
When using torch, while it is common to run the network in gpu, I wasn’t always so sure if it was mandatory for me to calculate the loss on the same gpu as the output at all times. If I can do loss calculation in cpu, then it would help Read more…
pytorch
torchvision fakedata example
torchvision supports various datasets and one of them is fakedata dataset. I was curious what this actually generated and here is an example the result is just an random RGB noised image.
deep learning
cross entropy loss / focal loss implmentation in pytorch
at the moment, the code is written for torch 1.4 binary cross entropy loss currently, torch 1.6 is out there and according to the pytorch docs, the torch.max function can receive two tensors and return element-wise max values. However, in 1.4 this feature is not yet supported and that is Read more…
pytorch
check cuda version build to torch package and find cudnn version used in torch
use the following python snippet to check cuda version the torch package was built against use the following python snippet to check cudnn version used by torch
pytorch
calculating gradient for selected tensors in pytorch
the above is an example code of showing how to calculate gradients for a few wanted tensors. In this case, I only wanted to calculate the gradient of conv2.weight so that I can later on update only this weight with the amount calculated based on the the gradient produced by Read more…
pytorch
avoiding full gpu memory occupation during training in pytorch
Problem While training even a small model, I found that the gpu memory occupation neary reached 100%. This seemed odd and it made me to presume that my pytorch training code was not handling gpu memory management properly. Here is a pseudo code for my pytorch training script. With this Read more…
pytorch
pytorch ckpt loading error due to device mismatch
Problem I have trained and saved model checkpoint which was train in GPU 1. However, when I try to load the checkpoint in GPU 0, it fails. The following is my ckpt loading code: this gives me the following error: Solution when loading ckpt data, ensure that device is mapped Read more…