@albanD
Any idea how to tweak code to make *mb to be list of three parameters like in previous versions of PyTorch 1.
m = Model007(vocab_size, n_fac).cuda()
ds = utils.TensorDataset(X, Y) 
dl = utils.DataLoader(ds, batch_size=2, shuffle=False) 
it = iter(dl)
*mb, yt = next(it)
y_hat = m(*mb)