Open lileidev opened 1 year ago
It reports "expected scalar type Double but found Float" with torch.mm in chapter 3.2.1. Fixed by add "to (torch.float32)" as bellow: features = torch.from_numpy(np.random.normal(0, 1, (num_examples, num_inputs))).to(torch.float32)
features = torch.from_numpy(np.random.normal(0, 1, (num_examples, num_inputs))).to(torch.float32)
It reports "expected scalar type Double but found Float" with torch.mm in chapter 3.2.1. Fixed by add "to (torch.float32)" as bellow:
features = torch.from_numpy(np.random.normal(0, 1, (num_examples, num_inputs))).to(torch.float32)