You’re welcome mate.
But you should be careful about two things:

If you want to use your code on GPU, you just need to create an instance of your model like this:

model = FC()
model.cuda()

I mean, you do not need to call  cuda  for  h0  in your  forward  method.

You should not pass  random  values as  h0  to your  GRU  object. If you do not want to initialize it (as before), pass zeros like this:

h0 = torch.zeros(2, 100, 128)

Good luck