-
## 🚀 Feature
Support torch::jit::script::Module converting to torch::nn::Module.
## Motivation
In some cases like image-style transfer, we need to calculate the gradient of loss w.r.t the input t…
-
I think the readme needs cleanup, ideas:
- [ ] update fancy examples gif, have some things from different domains. We also have axes now so something that demonstrates that! Maybe the 3D line stack…
-
### Describe the issue
Hi！I'm rather confused. I have trained two 2-layer CNN models. One is trained by Pytorch 1.13.0 and exported by torch.onnx. The other is trained by Tensorflow 1.10.0 and expo…
-
### Your current environment
```text
Collecting environment information...
PyTorch version: 2.2.1+cu121
Is debug build: False
CUDA used to build PyTorch: 12.1
ROCM used to build PyTorch: N/A
…
-
The task is to use a CNN model for a Matbench submission on regressing formation energy using the xtal2png representation (as an image and/or as an array would be fine). This will help with knowing ho…
-
I've been able to run regular Whisper just fine from a docker build, but haven't been able to get it to work for Faster Whisper
Currently using pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime as the …
-
Implement baseline with CNNs
- [x] Tensorflow
- [x] Pytorch
-
**Target:** Figure out if by default Torch `scaled_dot_product_attention` when executing in Hopper architecture, thus `PLATFORM_SUPPORTS_CUDNN_ATTENTION` and `SM90OrLater` evaluate to `True`, executes…
-
## Issue description
> RuntimeError: CUDA error: unspecified launch failure
Error occurring on any training script. Occurrence is not deterministic. Can occur at anytime during the course of tr…
-
### 🐛 Describe the bug
When trying to implement a custom `autograd.Function` wrapping a custom CUDA Extensions, I ran into what I believe is a bug or a poorly documented interaction.
A minimal e…