search

emited / VariationalRecurrentNeuralNetwork

Pytorch implementation of the Variational Recurrent Neural Network (VRNN).
282 stars 70 forks source link

The dim of the data #8

Open RuifMaxx opened 3 years ago

RuifMaxx commented 3 years ago

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 size of phi_x_t is ( batch_size, h_dim ), is that correct ?