-
I've been looking at training a smaller transformer s2s model. I prepared the am_transformer_s2s_librivox recipe and removed some transformer layers from the arch, and I'm seeing output like this for …
-
When I use _scatter_max()_ in my code, I run into some backward issue, which says the backward is not defined.
I wrote a sample code is here:
```python
import numpy as np
import torch
from t…
-
Hi,
In your SAC docs it says
> An alternate version of SAC, which slightly changes the policy update rule, can be implemented to handle discrete action spaces.
I was then wondering if you kne…
-
Congrats for the paper and thanks for the code! I have a question about how to use this method when there is no explicit GT for the sign(P).
Suppose that in the task of 3d pose estimation one is gi…
-
Hi there,
Just wondering how was the gumbel-softmax implemented in your scripts. Was a hard one-hot sample generated at each iteration of the searching phase, or a soft one? I noticed there's an op…
-
## 🐛 Bug
## To Reproduce
```
import torch
import torch.nn as nn
import math
if __name__ == "__main__":
batch_size = 128
temperature = 5.0
theta = torch.FloatTensor([1.75…
-
Sometimes nn.functional.gumbel_softmax will return nan if using GPU to calculate. It will not happen if using CPU to calculate.
test code:
```
import torch
import torch.nn as nn
import math
…
-
Hi, bjlkeng,
I read your blog and GitHub code. That's quite helpful for me as a new learner to VAE methods.
When I was reading the code of semi-VAE m2 model, I found that the unlabele…
-
Hey,
From my understanding of the paper, we are interested in the top k items in the random key vector that is generated by performing Logit(weights) + Gumbel(0,1). If this is correct, why is the s…
-
## 📚 Documentation
Docs for `torch.nn.functional.gumbel_softmax`.
Function `gumbel_softmax` docstring entry for `Gumbel-Softmax distribution` has two Arxiv links:
.. _Gumbel-Softmax di…
ZaydH updated
5 years ago