Open zainkhan-afk opened 3 months ago
I could not find a solution to this, so I just downloaded the tensorflow docker and ran the tensorflow notebook instead.
In my experience, working with the merlin-torch version is bad. Tensorflow has a much better support, I see the merlin-torch as still very immature. I had many problems in replicating the tensorflow-based notebooks with Merlin torch, as as you I've abandoned the idea and happily switched to merlin-tensorflow containers.
❓ Questions & Help
Details
I was running the Getting Started With MovieLens example for pytorch and when I created the dataloader and generated a single batch I got a different output than the the one that was expected:
I got:
Expected:
Docker: nvcr.io/nvidia/merlin/merlin-pytorch:nightly Notebook: 03-Training-with-PyTorch.ipynb
I read in the documentation that a multicoded object will have two tensors (value and nnzs), in my case I the two tensors are being added against different keys rather than being added as a tuple against a single key.
How can I get the batches in the required format?