-
On main branch (commit 08aa6cea0673fa97e43f5fa5ec09588e0af23d1b)
`python benchmarks/torchbench.py --inductor-settings -dcuda --float16 -n50 --inductor --filter timm_efficientdet` failed
```
Trace…
-
Dear author:
I am a student from China. I just only use the pretrained model to test the D4. When I enter the command" python3 test.py --model 1 --checkpoints ./D4-main/checkpoints/quick", The fo…
-
Dear all,
Has anyone succed run mobileetv3 small/large on Iphone with NPU/GPU/CPU accelerated？ in my case, i found the mobilenetv3 small run on CPU with 10ms inference time, compared with 20ms for …
-
I was using the UNet model with 'efficientnet-b2' encoder and 'imagenet' weights.
Here is a code snippet:
```
model = smp.Unet(encoder_name='efficientnet-b2', encoder_weights='imagenet', classes=…
-
## 🚀 Feature
_Note: To track the progress of the project check out [this board](https://github.com/pytorch/vision/projects/1)._
Add popular primitives (Losses, Schedulers, Data Augmentations, Op…
-
### 🐛 Describe the bug
The `permute` function works fine on CPU, but fails (with a warning) on MPS devices (on macOS):
```python
import torch
cpu = torch.device("cpu")
gpu = torch.device("mps…
-
### Is there an existing issue for this?
- [X] I have searched the existing issues and checked the recent builds/commits
### What happened?
receiving this error when loading deforum, they are sayin…
-
### 🐛 Describe the bug
I try to export Alexnet model to onnx
```python
import torch
from torchvision.models.alexnet import *
dummy_input = torch.randn(10, 3, 224, 224)
alexnet_model = alex…
-
## Description
I covert my onnx model into trt , but I get this error : `Error Code 4: Internal Error (Network must have at least one output)`
detail error is :
```
Loading ONNX file from path .…
-
## 🐛 Bug
Committed memory rises by about 1GB for each process spawned with either `multiprocessing.Process`, `torch.multiprocessing.Process` or `torch.multiprocessing.spawn` if `torch` is imported …