Hi, I want to create a lookup table or say a dictionary which is going to be accessed and modified during training. Using dict in Python, my lookup table is going to be saved in the CPU and I’m concerned about the back and forth between the GPU and CPU. How can I move the lookup table/dictionary from CPU to the GPU and avoid that? Can I use the nn.Embedding(..)  as a lookup table?
Thank you