All Questions
Tagged with pytorch torchvision
592 questions
0
votes
0
answers
41
views
Error while installing transformer_engine
[end of output]
note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for transformer_engine
Running setup.py clean for ...
0
votes
0
answers
48
views
Force TouchDesigner to use a specific python version
I'm running Windows10.
For my project I have to use Python 3.10 with tensorflow-directml, torchvision and other deep learning tools.
The following is a link to one of the many guides about this topic:
...
0
votes
1
answer
38
views
train and validation metrics and loss using torchvision
I am new to pytorch and torchvision.
Apparently, the __call__ method of a model returns different things if they are on train mode or not. The first case returns the loss, the second returns the ...
- 4,130
0
votes
1
answer
88
views
Equivalent of _new_empty_tensor from torchvision in pure torch function
I am trying to use a vision transformer repository (https://github.com/wangjian123799/L-DETR) which uses a version of torch and torchvision. Anyway, because exact versions are not given and torch is ...
- 5,723
0
votes
0
answers
25
views
Why py-torchvision fails in FresBSD 14 freshport package?
While trying to import:
import torchvision.models as models
I get the following runtime error:
---------------------------------------------------------------------------
RuntimeError ...
- 5,565
2
votes
1
answer
59
views
Does order of transforms applied for data augmentation matter in Torchvision transforms?
I have the following Custom dataset class for an image segmentation task.
class LoadDataset(Dataset):
def __init__(self, img_dir, mask_dir, apply_transforms = None):
self.img_dir = img_dir
...
- 33
0
votes
1
answer
133
views
Turn Grayscale mask-image into RGB while changing one grayscale into specific color
While using torch and matplotlib I was able to change the grayscale color #50 e.g. to red. However, only by saving the image inbetween.
How can it be solved without saving the image?
I am sorry for ...
- 1
1
vote
1
answer
57
views
OutOfMemoryError: CUDA out of memory with a gan
I have been trying to setup this gan but i cannot escape this error: https://pastebin.com/rp0kNXiH (sometimes it's not the same line)
I have tried gc.collect but maybe i misplaced it. I also tried to ...
- 13
1
vote
0
answers
76
views
cannot import name 'OpOverload' from 'torch._ops'
when i run a cell contain
from torchvision import datasets,transforms
and an error appears.
---------------------------------------------------------------------------
ImportError ...
- 11
0
votes
0
answers
22
views
Celery tasks and Torch multiprocessing
I have a script with neural networks here is example of init
import torch
from torchvision.models.detection import fasterrcnn_resnet50_fpn
from torchvision.models.detection import ...
0
votes
0
answers
43
views
Count number of trainable parameters from model start to given layer
I have a number of pytorch models and would like to print the number of trainable parameters it took to get to specific points in the feed forward function.
I know how get the number of all trainable ...
- 11
0
votes
1
answer
39
views
What should be my classifier in Transfer Learning using MaxViT?
I am trying to do transfer learning on Pytorch pretrained models with custom dataset. I have been able to successfully perform transfer learning with SqueezeNet.
For Squeezenet my classifier was, ...
- 3
0
votes
0
answers
120
views
TypeError: empty() received an invalid combination of arguments
I want to do object detection using cascade r-cnn and swin transformer. Here is my code and the libraries I installed.
mmcv 2.1.0
mmdet 3.3.0 ...
0
votes
0
answers
492
views
Is torch2.4.1 compatible with CUDA 12.5?
I locally installed my CUDA Toolkit 12.5. And everything went well (I can successfully call torch.cuda.is_available() and returns true) before I install torchvision with cmd conda install torchvision -...
0
votes
0
answers
65
views
Compatibility Issue with torch_scatter
I use Torch 2.4.1, Torchvision 0.19.1, and CUDA 11.8 for my model. Torch-scatter 2.12, which I have already installed, is incompatible with these. How can I determine which version is suitable?
I want ...
