Category "pytorch"

Optimising model.parameters and custom learnable parameter together using torch.optim gives non-leaf tensor error

Framework: PyTorch I am trying to optimise a custom nn.parameter(Temperature) used in softmax calculation along with the model parameters using a single Adam op

How can I determine validation loss for faster RCNN (PyTorch)?

I followed this tutorial for object detection: https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html and their GitHub repository that contains th

Automatically check available GPU on Google Colab

Is there an automatic way to check which GPU is currently available on Google Colab (Pro). Say I would like to use a Tesla P100 instead of the Tesla T4 to train

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