Hi, I created a custom pytorch module in C. Currently it only supports cpu, not cuda. I wish to know what happens when I use this module in a model and do model.cuda(). Does model.cuda() keep the parameters of this module in cpu.
It seems to be working fine though.