Your weights and bias seem to have a wrong shape, since the default linear layer using your setup would use:
print(nn.Linear(50, 78).weight.shape)
> torch.Size([78, 50])
print(nn.Linear(50, 78).bias.shape)
> torch.Size([78])

so you migth need to fix it (permute the weights and make sure the bias has a single dimension).