-
CenterCrop is not traceable
```python
from torchvision.transforms import CenterCrop
c = CenterCrop(size=(112, 112))
torch.jit.trace(c, example_inputs=torch.ones((200, 200)))
```
```bash
Typ…
-
```
Traceback (most recent call last):
File "train.py", line 537, in
train(hyp, opt, device, tb_writer, wandb)
File "train.py", line 336, in train
results, maps, times = test.test(op…
-
I understood that "restart_prob" effect random walk length and "max_nodes_per_seed" effect trace numbers.
I tested "random_walk_with_restart" with many different parameters.
I notice that "restart_…
-
### Tasks:
- [ ] Make autocast ops extensible, e.g. via a function that modifies a global list of ops & their behavior, **per device**
- [ ] To better copy eager behavior, we could try using codegen…
-
## Bug Description
I met a bug when I use torch-tensorrt to quantize a YOLOv6 model into fp8_e4m3. I use the code in [vgg_fp8_code](https://github.com/pytorch/TensorRT/blob/main/examples/dynamo/vg…
-
Hi @gahdritz,
Here are a bit more details about my error when adding --trace_model option while doing inferencing.
Here is the commands I run:
```
python3 run_pretrained_openfold.py \
/mnt…
-
## 🐛 Bug
The output from a traced model using (1) python and (2) c++ are different.
xpost: https://discuss.pytorch.org/t/c-and-pytorch-inference-discrepancy/77388?u=jonrbates
## To Repr…
-
`gcp_opt` doesn't track timings internally when LBFGS is the optimizer. Matlab has this (only using a simple tic/toc outside the call to the C code/mex). [See here](https://gitlab.com/tensors/tensor_t…
-
### 🐛 Describe the bug
I am following this example code to save and load optimizer states using load_sharded_optimizer_state_dict: https://github.com/pytorch/pytorch/blob/c75e064dd6a2f800476bc84d4f…
-
### Enhancement Request - Support for Additional LLM Types
#### Description:
After reviewing the [Mediapipe documentation](https://developers.google.com/mediapipe/solutions/genai/llm_inference) an…