In the documentation for torch.rot90 it is stated that Rotation direction is from the first towards the second axis if k > 0, and from the second towards the first for k < 0. However say that we are rotating from axis 0 to axis 1, is axis 0 rotating to axis 1 in the ..
I am trying to do demo runs using yolo v5 Yolo V5 However in my demo case, for example, i don’t need to track people, only cars. Is there any way to set tracing classes for yolo v5? I searched through different topics and the only solution i found is to train on custom dataset. ..
I have a tensor with this size torch.Size([128, 64]) how do I add one "dummy" dimension such as torch.Size([1, 128, 64]) Source: Python..
For experimental reasons, I am trying to change the values inside a tensor of size (64,1,28,28) which is a tensor containing the pixel values of 64 images from the MNIST dataset. I am trying to replace them all with ones. However, I am trying to apply the where method of torch with no avail. main ..
I am running the Ubuntu 18.04 Deep Learning AMI (DLAMI) on AWS, and am attempting to run it on a p2.xlarge EC2 instance, but cannot get CUDA working. torch.cuda.is_available() is returning false in ipython, even though nvidia-smi is showing that I have a GPU available: [email protected]:~$ source activate pytorch_latest_p37 (pytorch_latest_p37) [email protected]:~$ ipython Python 3.9.5 (default, ..
For the last couple of weeks, I have been really struggling to debug this new problem when training my detection model. Couple of weeks before the training worked great, but now all of a sudden I get this new unheard error. I make my model and train overall like this but still get an error. ..
I am learning the best ways to manage batches and other best practices during model training and inference and I have some: If I have a batch that I move to GPU, should. I move it back to CPU after doing the training? If no, why? batch, label = batch.to(device), label.to(device) model(batch) # ..Training pass… ..
I was using the torch.tensor.repeat() x = torch.tensor([[1, 2, 3], [4, 5, 6]]) period = x.size(1) repeats = [1,2] result = x.repeat(*repeats) the result is tensor([[1, 2, 3, 1, 2, 3], [4, 5, 6, 4, 5, 6]]) if I get result as follows result = x.repeat(repeats) the result is the same tensor([[1, 2, 3, 1, ..
I’m actually working with a CNN network for image classification. what I’m trying to do is to add an external function to that network , more precisely after the convolutional layers. the function was built for grayscale images so it accept as input tensor with shape (m, n), and the out put of the convolutional ..
I am trying to install Torch in a Google Colab notebook, and after running the following cell: %cd /content/ !git clone https://github.com/nagadomi/distro.git torch –recursive import os os.chdir(‘./torch/’) !bash install-deps !./install.sh !. ./install/bin/torch-activate it gets stuck on !./install.sh with the error /content/torch/extra/cutorch/init.c:938:34: error: ‘struct cudaPointerAttributes’ has no member named ‘isManaged’ after trying to read this line ..