-
Notifications
You must be signed in to change notification settings - Fork 68
Open
Description
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 ?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels