Category "pytorch"

Pytorch parallel functional calls for multiple locations in a tensor

Does anyone know if there is a function in PyTorch that allows you to call a particular function for all the locations in a tensor where a condition is satisfie

Tracing Tensor Sizes in TorchScript

I'm exporting a PyTorch model via TorchScript tracing, but I'm facing issues. Specifically, I have to perform some operations on tensor sizes, but the JIT compi

How define the number of class in Detectron2 bounding box predict Pytorch?

Where should i define the number os classes ? ROI HEAD or RETINANET ? Or both should have the same value ? cfg.MODEL.RETINANET.NUM_CLASSES =int( len(Classe_list

Flexible word count in Pytorch Embedding

The Embedding class in Pytorch takes in num_embedding as a parameter. According to the doc, num_embedding is "size of the dictionary of embeddings". I am curiou

PyTorch XLA incompatibility issue when running a modified official example

When running this sample using PyTorch XLA modified slightly from the official sample in Colab, the following errors are presented: ImportError: /usr/local/lib/

Training a u-net for multi-landmark heatmap regression producing the same heatmap for each channel

I’m training a U-Net (model below) to predict 4 heatmaps (gaussian centered around a keypoint, one in each channel). Each channel is for some reason outpu

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

Can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

I am trying to show results of GAN network on some specified epochs. The function for printing the current result was used previously with TF. I need to change

ai.djl.engine.EngineException: PytorchStreamReader failed locating file constants.pkl: file not found

I tried to use DJL to load the Yolov5 model for target recognition, but an error was reported. ai.djl.engine.EngineException: PytorchStreamReader failed locatin

Tacotron2 traning new languages for speech synthesis using Pytorch

I wanted to see if it's possibe to train the Tacotron2 model for languages other than English (LJ Speech Dataset) using Pytorch. If so, how do I train the model

Cuda illegal memory access when running inference on *.engine

After exporting a YoloV5 model to .engine I receive an error when trying to perform inference on it. Loading model.engine for TensorRT inference... [01/16/2022-