def to_gpu(self): super(VAELM, self).to_gpu() self.encoder.to_gpu() self.decoder.to_gpu() for i in range(self.num_layers): self.get_htrans(i).to_gpu() self.get_ctrans(i).to_gpu()