VariationalRecurrentNeuralNetwork
VariationalRecurrentNeuralNetwork copied to clipboard
emited
Pytorch implementation of the Variational Recurrent Neural Network (VRNN).
the code ` def _nll_bernoulli(self, theta, x): return - torch.sum(x*torch.log(theta + EPS) + (1-x)*torch.log(1-theta-EPS))` may got nan loss. i think it should be `def _nll_bernoulli(self, theta, x): return - torch.sum(x*torch.log(theta...
When load the data from the dataloader, the dim is transposed: `data = Variable(data.squeeze().transpose(0, 1)).to(device)` So in model.py line 82 ``` for t in range(x.size(0)): phi_x_t = self.phi_x(x[t]) ``` the...
Error message : Done! Traceback (most recent call last): File "train.py", line 112, in train(epoch) File "train.py", line 25, in train data = (data - data.min().data[0]) / (data.max().data[0] - data.min().data[0])...
Train Epoch: 3 [0/60000 (0%)] KLD Loss: 2.687659 NLL Loss: 73.599564 Train Epoch: 3 [2800/60000 (21%)] KLD Loss: 2.976363 NLL Loss: 78.757454 Train Epoch: 3 [5600/60000 (43%)] KLD Loss: 2.837864...
Hi, Can you attach a license for the code so it's easier for others to reuse? Thank you very much.
@emited Please check and let me know. 1. Added clipping before the optimizer step. 2. Updated with .item() rather than .data[0] 3. Input is normalized but the reconstruction is done...
