Not work.
I use jupyter notebook. Running cell of class Model(), it calls the “q_sel” function of;
File "QNet.py", line 63, in q_sel
    self.sel[index] = torch.sigmoid(self.w_x[index] * self.x[index] + self.w_h[index] * self.h[index])

Then makes the error.