@ptrblck
No error occurs but the model runs on CPU. To upload h_in and c_in to GPU, what should I do among 3 possibilities?
h_in = nn.Parameter(torch.randn(self.num_layer, batch_size, self.hidden_dim)).cuda()
h_in = nn.Parameter(torch.randn(self.num_layer, batch_size, self.hidden_dim).cuda())
h_in = torch.randn(self.num_layer, batch_size, self.hidden_dim).cuda()