Hi,
The problem is that you store you convolutions in a plain python list. This means that .to() cannot find these convs to move them to cuda.
You should do self.convs = nn.ModuleList() instead of self.convs = [] and that will work fine.